flywire-gnn

校正済みのメス全脳コネクトームを、そのまま学習に使えるグラフデータセットとして整えたパッケージ。ニューロン 139,255 個、シナプス 5 本以上の有向ペア 2,700,513 本、ノードごとに次数・79 領域の神経網別プロファイル・神経伝達物質プロファイルから作った 174 次元の特徴量、エッジ属性 7 次元、全ノードを覆う 9 クラスのラベルを持つ。クラス 1 つでダウンロードとキャッシュが済み、決定論的な層化分割と再現可能なベースライン 3 本が付く。課題は「配線の構造だけからニューロンの種別を当てる」こと。

公式サイト ↗

#pytorch-geometric#dataset#node-classification#baselines#flywire