def load_ch_matrix(): print "Reading NodeID map started at", time.strftime(time_format) start = time.time() node_id_map = inout.load_from_file(path + "{0}_node_id_map".format(dataset)) print "Reading NodeID map took", time.time() - start, "s" print "-----------------------------------------" print "Reading hypergraph started at", time.strftime(time_format) start = time.time() hypergraph = Hypergraph.load_from_file(path + "{0}_hgraph".format(dataset)) print "Reading hypergraph took", time.time() - start, "s" print "-----------------------------------------" print "Reading characteristic matrix started at", time.strftime( time_format) start = time.time() ch_matrix = CharacteristicMatrix.load_from_file( path + "{0}_ch_matrix".format(dataset)) print "Reading characteristic matrix took", time.time() - start, "s" print "-----------------------------------------" print "Reading Column index to Node map started at", time.strftime( time_format) start = time.time() index_node_map = inout.load_from_file(path + "{0}_index_node_map".format(dataset)) print "Reading Column index to Node map took", time.time() - start, "s" print "-----------------------------------------" return ch_matrix, hypergraph, index_node_map, node_id_map
def load_ch_matrix(): print "Reading NodeID map started at", time.strftime(time_format) start = time.time() node_id_map = inout.load_from_file(path + "{0}_node_id_map".format(dataset)) print "Reading NodeID map took", time.time() - start, "s" print "-----------------------------------------" print "Reading hypergraph started at", time.strftime(time_format) start = time.time() hypergraph = Hypergraph.load_from_file(path + "{0}_hgraph".format(dataset)) print "Reading hypergraph took", time.time() - start, "s" print "-----------------------------------------" print "Reading characteristic matrix started at", time.strftime(time_format) start = time.time() ch_matrix = CharacteristicMatrix.load_from_file(path + "{0}_ch_matrix".format(dataset)) print "Reading characteristic matrix took", time.time() - start, "s" print "-----------------------------------------" print "Reading Column index to Node map started at", time.strftime(time_format) start = time.time() index_node_map = inout.load_from_file(path + "{0}_index_node_map".format(dataset)) print "Reading Column index to Node map took", time.time() - start, "s" print "-----------------------------------------" return ch_matrix, hypergraph, index_node_map, node_id_map
def testCharacteristicMatrix_ReadWrite(self): file_name = "test_files/characteristic_matrix.tmp" dummy_hypergraph = Hypergraph(example_graphs.snm_dummy_graph) rballs_database, _ = similar_nodes_mining.extract_rballs_database(dummy_hypergraph, r_in=2, r_out=2, r_all=0) nodes_count = dummy_hypergraph.number_of_nodes() ch_matrix = CharacteristicMatrix(rballs_database, nodes_count, wl_iterations=4) ch_matrix.save_to_file(file_name) read_ch_matrix = CharacteristicMatrix.load_from_file(file_name) self.assertEqual(read_ch_matrix, ch_matrix, "The read characteristic matrix is different from the saved one.")
def testCharacteristicMatrix_ReadWrite(self): file_name = "test_files/characteristic_matrix.tmp" dummy_hypergraph = Hypergraph(example_graphs.snm_dummy_graph) rballs_database, _ = similar_nodes_mining.extract_rballs_database( dummy_hypergraph, r_in=2, r_out=2, r_all=0) nodes_count = dummy_hypergraph.number_of_nodes() ch_matrix = CharacteristicMatrix(rballs_database, nodes_count, wl_iterations=4) ch_matrix.save_to_file(file_name) read_ch_matrix = CharacteristicMatrix.load_from_file(file_name) self.assertEqual( read_ch_matrix, ch_matrix, "The read characteristic matrix is different from the saved one.")