Skip to content

Latest commit

 

History

History
22 lines (11 loc) · 511 Bytes

README.md

File metadata and controls

22 lines (11 loc) · 511 Bytes

基于不变信息的分类(最大化互信息损失函数)

目录组织结构

  1. Nets:存储模型网络结构代码的目录

  2. Models:存储训练模型权重的目录

  3. DataLoaders:存储加载数据集代码的目录

  4. DataTransers:存储数据变换方法代码的目录

其他的还有单独的文件:

  1. IIC_Loss.py:最大化互信息的损失函数

  2. train.py:训练代码

  3. val.py:对模型权重进行验证的代码

  4. config.py:对可变参数进行配置