third_codes/eland
1
0
Fork 0
mirror of https://github.com/elastic/eland.git synced 2025-07-11 00:02:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add number_samples to sklearn MLModel

Browse Source
This commit is contained in:
P. Sai Vinay 2021-10-07 18:44:54 +05:30 committed by GitHub
parent 014943d3b8
commit f9d2defb1b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
eland/ml/transformers/sklearn.py
View File

@ -92,6 +92,7 @@ class SKLearnTransformer(ModelTransformer):
node_index, node_index,
decision_type=self._node_decision_type, decision_type=self._node_decision_type,
leaf_value=leaf_value, leaf_value=leaf_value,
number_samples=int(node_data[5]),
) )
else: else:
return TreeNode( return TreeNode(
@ -101,6 +102,7 @@ class SKLearnTransformer(ModelTransformer):
right_child=int(node_data[1]), right_child=int(node_data[1]),
split_feature=int(node_data[2]), split_feature=int(node_data[2]),
threshold=float(node_data[3]), threshold=float(node_data[3]),
number_samples=int(node_data[5]),
) )