文件名称:comet:[ICLR 2021]少量学习的概念学习者
文件大小:715KB
文件格式:ZIP
更新时间:2024-06-12 22:16:27
Python
少量学习的概念学习者 凯迪(Kaidi Cao)*,玛丽亚(MariaBrbić)*,尤里(Jure Leskovec) 此存储库包含COMET算法的PyTorch中的参考源代码。 COMET是一种元学习方法,可沿人类可理解的概念维度学习可概括的表示形式。 有关更多详细信息,请查阅我们的论文 (ICLR '21)。 依存关系 该代码使用以下库构建: 1.5 json 入门 CUB数据集 将目录更改为./filelists/CUB 运行source ./download_CUB.sh Tabula Muris数据集 将目录更改为./filelists/tabula_muris 运行source ./download_TM.sh 用法 训练 我们在这里提供一个示例: 运行python ./train.py --dataset CUB --model Conv4NP --method
【文件预览】:
comet-master
----.gitignore(2KB)
----CUB()
--------data()
--------save_features.py(5KB)
--------utils.py(1KB)
--------test.py(8KB)
--------methods()
--------backbone.py(15KB)
--------io_utils.py(5KB)
--------filelists()
--------configs.py(265B)
--------train.py(10KB)
----LICENSE(19KB)
----README.md(3KB)
----TM()
--------data()
--------save_features.py(4KB)
--------utils.py(1KB)
--------test.py(6KB)
--------methods()
--------backbone.py(17KB)
--------io_utils.py(4KB)
--------filelists()
--------configs.py(265B)
--------train.py(7KB)
----img()
--------COMET_model.png(648KB)