Skip to content

Commit b8e36b8

Browse files
author
[zebinyang]
committed
add feature importance for each SIM; get_projection_equation; add get_feature_importance; add get_projection_index; update visualize_one_leaf; add update version 0.1.5
1 parent f20d0b8 commit b8e36b8

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

simtree/simtree.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -123,7 +123,7 @@ def get_projection_equation(self, node_id, precision=3):
123123
equation = ""
124124
importance = self.get_feature_importance(node_id)
125125
sortind = np.argsort(importance)[::-1]
126-
est = self.leaf_estimators_[node_id].
126+
est = self.leaf_estimators_[node_id]
127127
for i in range(est.beta_.shape[0]):
128128
if i == 0:
129129
equation += str(round(np.abs(est.beta_[sortind[i], 0]), 3)) + clf.feature_names[sortind[i]]

0 commit comments

Comments
 (0)