def constructor(self, params, recalculate = False, to_pickle = False, to_filelize = False, always_recalculate = False): pdb_names = self.get_param(params, 'pdb_names') chain_letters = self.get_param(params, 'chain_letters') for i in range(len(pdb_names)): pdb_name = pdb_names[i] chain_letter = chain_letters[i] self.set_param(params, 'pdb_name', pdb_name) self.set_param(params, 'chain_letter', chain_letter) the_folder = constants.BIN_FOLDER + pdb_name + '/' if not os.path.exists(the_folder): os.makedirs(the_folder) pdb.set_trace() node_features = self.get_var_or_file(objects.bmW, params, False, False, False) transposed_node_features = helper.get_transpose(node_features) pdb.set_trace() helper.write_mat(transposed_node_features, the_folder + 'Xnode.csv') pdb.set_trace() edge_features = self.get_var_or_file(objects.bnW, params, False, False, False) edge_features_transposed = helper.get_transpose(edge_features) helper.write_mat(edge_features_transposed, the_folder + 'Xedge.csv') true_states = self.get_var_or_file(objects.oW, params, False, False, False) true_states = [true_states[i]+1 for i in range(len(true_states))] helper.write_vect(true_states, the_folder + 'true_y.csv') edge_list = self.get_var_or_file(objects.iW, params, False, False, False) helper.write_mat(edge_list, the_folder + 'edge_list.csv') num_nodes = len(node_features) adj_mat = [ [0 for i in range(num_nodes)] for j in range(num_nodes)] for i in range(len(edge_list)): n1 = edge_list[i][0] n2 = edge_list[i][1] adj_mat[n1][n2] = 1 adj_mat[n2][n1] = 1 helper.write_mat(adj_mat, the_folder + 'adj_mat.csv') info = [str(len(node_features)), str(len(edge_features)), str(2), str(len(node_features[0])), str(len(edge_features[0]))] helper.write_vect(info, the_folder + 'info.txt', the_sep = ' ') return None
def dump_object(self, obj): write_vect(obj, self.get_holding_location())