Commit 40c698dd authored by zzhang-cn's avatar zzhang-cn
Browse files

list alias

parent a347214f
......@@ -88,7 +88,7 @@ class TreeGlimpsedClassifier(NN.Module):
{i: 'h%d' % i for i in range(len(G.nodes()))},
False
)
self.h_nodes_list = h_nodes_list = list(G.nodes)[:]
self.h_nodes_list = h_nodes_list = list(G.nodes)
for h in h_nodes_list:
G.node[h]['type'] = 'h'
b_nodes_list = ['b%d' % i for i in range(len(h_nodes_list))]
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment