Github 上地址 github.com/thuml/TimesNet. 的开源项目 TIMESNET 在 `data_loader.py` 中 class Dataset_Custom(Dataset) 的初始化函数 __init__ () 每个输入参数的具体含义

• 日期: 2024-09-06
• 标签: 科技

1. csv_file_path: 数据集的 CSV 文件路径。
2. img_dir: 存储图片文件的目录路径。
3. transform: 数据集图片的预处理转换函数。
4. mode: 数据集模式，有 "train" 和 "test" 两种模式。
5. img_size: 图片的尺寸大小。
6. crop_size: 图片的裁剪尺寸。
7. num_classes: 数据集的类别数量。
8. label_file_path: 数据集标签文件的路径。
9. use_balanced_sampler: 是否使用平衡采样器。
10. balance_factor: 平衡采样器的采样因子。
11. use_mixup: 是否使用 Mixup 数据增强方法。
12. mixup_alpha: Mixup 数据增强方法的 alpha 参数。

原文地址: https://www.cveoy.top/t/topic/yI5 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录