Commitfd5fc82

committed

Fix as_tensor

1 parent96eb50e commitfd5fc82Copy full SHA for fd5fc82

File tree

-8

lines changed

-8

lines changed

Lines changed: 9 additions & 8 deletions

Original file line number	Diff line number	Diff line change
`@@ -64,7 +64,10 @@ def __init__(self, i, atts):`
`64`	`64`	`self.nodes_missing_value_tracks_true=None`
`65`	`65`	`fork,vinatts.items():`
`66`	`66`	`ifk.startswith("nodes"):`
`67`		`-setattr(self,k,v[i])`
	`67`	`+ifk.endswith("_as_tensor"):`
	`68`	`+setattr(self,k.replace("_as_tensor",""),v[i])`
	`69`	`+else:`
	`70`	`+setattr(self,k,v[i])`
`68`	`71`	`self.depth=0`
`69`	`72`	`self.true_false=""`
`70`	`73`	`self.targets= []`
`@@ -120,10 +123,7 @@ def process_tree(atts, treeid):`
`120`	`123`	`]`
`121`	`124`	`fork,vinatts.items():`
`122`	`125`	`ifk.startswith(prefix):`
`123`		`-if"classlabels"ink:`
`124`		`-short[k]=list(v)`
`125`		`-else:`
`126`		`-short[k]= [v[i]foriinidx]`
	`126`	`+short[k]=list(v)if"classlabels"inkelse [v[i]foriinidx]`
`127`	`127`
`128`	`128`	`nodes=OrderedDict()`
`129`	`129`	`foriinrange(len(short["nodes_treeids"])):`
`@@ -132,9 +132,10 @@ def process_tree(atts, treeid):`
`132`	`132`	`foriinrange(len(short[f"{prefix}_treeids"])):`
`133`	`133`	`idn=short[f"{prefix}_nodeids"][i]`
`134`	`134`	`node=nodes[idn]`
`135`		`-node.append_target(`
`136`		`-tid=short[f"{prefix}_ids"][i],weight=short[f"{prefix}_weights"][i]`
`137`		`- )`
	`135`	`+key=f"{prefix}_weights"`
	`136`	`+ifkeynotinshort:`
	`137`	`+key=f"{prefix}_weights_as_tensor"`
	`138`	`+node.append_target(tid=short[f"{prefix}_ids"][i],weight=short[key][i])`
`138`	`139`
`139`	`140`	`defiterate(nodes,node,depth=0,true_false=""):`
`140`	`141`	`node.depth=depth`

Comments

(0)