Skip to content
Snippets Groups Projects
Commit f59ccdcd authored by Ali Karimi's avatar Ali Karimi
Browse files

not save the partition

parent 09262f58
No related branches found
No related tags found
No related merge requests found
......@@ -10,7 +10,7 @@ import community as community_louvain
from brainNetwork.util.path import get_local_datadir
csv_dir = os.path.join(get_local_datadir(),'connectivity_paper.csv')
csv_dir = os.path.join(get_local_datadir(),'connectivity_tanh.csv')
connect_mat = pd.read_csv(csv_dir, index_col=0)
# Get the graph
G = nx.convert_matrix.from_pandas_adjacency(connect_mat)
......@@ -20,7 +20,8 @@ print(f"The modularity is: {community_louvain.modularity(partition_louvian,G)}")
# Make the dict into a data frame
partition_pd = pd.DataFrame(partition_louvian, index = ['partition'])
assert all(partition_pd.columns == connect_mat.columns)
partition_pd.to_csv(os.path.join(get_local_datadir(), 'partition.csv'))
# Write to CSV file
#partition_pd.to_csv(os.path.join(get_local_datadir(), 'partition.csv'))
# reorder columns and rows
reorder_indices = partition_pd.columns[np.squeeze(np.argsort(partition_pd.values))]
connect_mat = connect_mat.reindex(columns = reorder_indices, index = reorder_indices)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment