from time import time import ProcessData import BuildingGraph import networkx as nx t0 = time() graphs = [] train_labels, train_sents = ProcessData.PreprocessFile( './DataSet/r8-train-no-stop.txt') test_labels, test_sents = ProcessData.PreprocessFile( './DataSet/r8-test-no-stop.txt') i = 0 for doc in train_sents: G = BuildingGraph.builder(directed=False, weighted=False, window_size=4, inputDoc=doc) graphs.append(G) nx.write_gexf(G, './R8/R8_nostop_graphgefx/docs_' + str(i) + '.train.gexf') i += 1 print("Time for building all graphs: "), time() - t0 print("Number of graphs: "), len(graphs)