关于使用其他数据集训练的问题

aigc-apps / AMFormer

The AMFormer algorithm, accepted at AAAI-2024, for deep tabular learning

GNU General Public License v3.0

25 stars 7 forks source link

Closed ZZC2002ZZC closed 5 months ago

ZZC2002ZZC commented 6 months ago

尊敬的作者您好：我想尝试用您的模型进行多分类任务，该怎么运作。

Ch3ngY1 commented 6 months ago

config/datasets/中的yaml文件里有个out参数，把out参数设置成你最终的分类结果数量就能完成多分类，然后根据数据集设置对应的categories即可

ZZC2002ZZC commented 6 months ago

谢谢您的回复，想问下在哪里加载数据集的路径呢

---原始邮件--- 发件人: @.> 发送时间: 2024年4月24日(周三) 中午11:34 收件人: @.>; 抄送: @.**@.>; 主题: Re: [aigc-apps/AMFormer] 关于使用其他数据集训练的问题 (Issue #2)

config/datasets/中的yaml文件里有个out参数，把out参数设置成你最终的分类结果数量就能完成多分类，然后根据数据集设置对应的categories即可

— Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you authored the thread.Message ID: @.***>

Ch3ngY1 commented 6 months ago

可以参考utils/data_load/data_hcdr.py，这边写的是固定的路径，因为不同数据集的处理方式不同，就直接写在data_load模块中了

jianghai1999 commented 6 months ago

尊敬的作者您好：不好意思，打扰您了，请问一下，如果用来做回归预测，除了修改out参数输出为1以外，还需要修改那些参数呢？

Ch3ngY1 commented 6 months ago

@jianghai1999 根据你具体的数据集设置categories，这部分主要是为了把类别数据用于之后的embedding，如果你已经完成了embedding可以不管这一部分，直接输入到模型的特征提取部分就行了

waterha commented 2 months ago

作者您好，请问源码中data_hcdr.py中 class make_split()模块中的file 和 root 代表什么