[FIX] Naming of nodes in generator (#85, #90)

Author: diviyank

cdt/data/acyclic_graph_generator.py

@@ -173,7 +173,7 @@ def generate(self, rescale=True):
             if rescale:
                 self.data['V{}'.format(i)] = scale(self.data['V{}'.format(i)].values)
 
-        return self.data, self.g
+        return self.data, nx.relabel_nodes(self.g, {i:'V'+str(i) for i in self.g.nodes}, copy=True)
 
     def to_csv(self, fname_radical, **kwargs):
         """

requirements.txt

@@ -4,7 +4,7 @@ scikit-learn>=0.21.1
 joblib>=0.13.2
 pandas>=0.24.1
 networkx>=2.3
-torch==1.4.0
+torch
 tqdm>4.0.0
 GPUtil>=1.4.0
 statsmodels>=0.9.0