DataFram данных графика в Stellargraph

      from scipy.io import loadmat
import pandas as pd
import networkx as nx
import stellargraph as sg

path = 'circuit-data/for tinghao/template1-lib5-eqns-CR-RESULTS-SET1-FINAL.mat'
raw_data = loadmat(path, squeeze_me=True)
data = raw_data['Graphs']

A = data['A']
nz = data['nz']
np = data['np']
graphs = []
node_degrees = []
s_graphs = []
graph_data = pd.DataFrame()

for i in range(len(A)):
    graphs.append(nx.Graph(A[i]))

for i in range(len(graphs)):
    node_degrees.append(nx.degree(graphs[i]))

graph_data['node_degrees'] = node_degrees
graph_data['nz'] = nz
graph_data['np'] = np

for i in range(len(graphs)):
    s_graphs.append(sg.StellarGraph.from_networkx(graphs[i]), node_features=graph_data)

Я намерен создать свою собственную базу данных графиков с функциями узлов. Затем используйте их все сразу в машинном обучении с помощью stellargraph.

У меня вопрос, когда я бегу s_graphs.append(sg.StellarGraph.from_networkx(graphs[i], node_features=graph_data)), будет ли каждая строка в graph_data назначена соответствующим графам?

0 ответов

Другие вопросы по тегам