data.py 文件源码

def _load_data(self):
        # adjacency matrix
        adj = np.loadtxt(self.data, delimiter = ",")
        self._ref = nx.DiGraph()
        self._ref = nx.from_numpy_matrix(adj, create_using = self._ref)

        # vertex attributes
        with open(self.details, mode = "r") as handle:
            read = csv.reader(handle)
            for row in read:
                u = self._ref.node[int(row[0])]
                u["ndd"] = row[1] == "1"
                u["bp"] = row[2]
                u["bd"] = row[3]