Commit8382877

committed

improve tree plotting

1 parent5ce9398 commit8382877Copy full SHA for 8382877

File tree

+15

-3

lines changed

+15

-3

lines changed

Lines changed: 12 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -65,6 +65,18 @@ def test_onnx_text_plot_tree_cls(self):`
`65`	`65`	`self.assertIn(" T y=",res)`
`66`	`66`	`self.assertIn("n_classes=3",res)`
`67`	`67`
	`68`	`+deftest_onnx_text_plot_tree_cls_2(self):`
	`69`	`+iris=load_iris()`
	`70`	`+X_train,y_train=iris.data.astype(numpy.float32),iris.target`
	`71`	`+clr=DecisionTreeClassifier()`
	`72`	`+clr.fit(X_train,y_train)`
	`73`	`+model_def=to_onnx(`
	`74`	`+clr,X_train.astype(numpy.float32),options={"zipmap":False}`
	`75`	`+ )`
	`76`	`+res=onnx_text_plot_tree(model_def.graph.node[0])`
	`77`	`+self.assertIn("n_classes=3",res)`
	`78`	`+print(res)`
	`79`	`+`
`68`	`80`	`@ignore_warnings((UserWarning,FutureWarning))`
`69`	`81`	`deftest_onnx_simple_text_plot_kmeans(self):`
`70`	`82`	`x=numpy.random.randn(10,3)`

Lines changed: 3 additions & 3 deletions

Original file line number	Diff line number	Diff line change
`@@ -114,9 +114,9 @@ def process_tree(atts, treeid):`
`114`	`114`	`foriinrange(len(short[f"{prefix}_treeids"])):`
`115`	`115`	`idn=short[f"{prefix}_nodeids"][i]`
`116`	`116`	`node=nodes[idn]`
`117`		`-node.target_nodeids=idn`
`118`		`-node.target_ids=short[f"{prefix}_ids"][i]`
`119`		`-node.target_weights=short[f"{prefix}_weights"][i]`
	`117`	`+node.append_target(`
	`118`	`+id=short[f"{prefix}_ids"][i],weight=short[f"{prefix}_weights"][i]`
	`119`	`+)`
`120`	`120`
`121`	`121`	`defiterate(nodes,node,depth=0,true_false=""):`
`122`	`122`	`node.depth=depth`

Comments

(0)