Skip to content

Latest commit

 

History

History
22 lines (13 loc) · 680 Bytes

readme.md

File metadata and controls

22 lines (13 loc) · 680 Bytes

Transformer-based Hierarchical Clustering

This repository contains the code for the paper "Transformer-Based Hierarchical Clustering for Brain Network Analysis," to be published at IEEE-ISBI 2023.

Dataset

ABIDE can be accessed from here. The preprocessing scripts can be found at util/abide, with instructions in the same folder.

Usage

ABIDE

python main.py --config_filename setting/abide_dec_ortho_learnable_newpool_sftmx_topk_90_4_hierarchical_sum_5-4lr.yaml

ABCD

python main.py --config_filename setting/abcd_dec_ortho_learnable_newpool_sftmx_topk_90_4_hierarchical_sum.yaml