许可证：apache-2.0
标签：

• 训练生成
  数据集：
• 图像文件夹
  评估指标：
• 准确率
• 精确率
• 召回率
• F1分数

模型索引：

• 名称：resnet-18特征提取
  结果：
  • 任务：
    名称：图像分类
    类型：image-classification
    数据集：
    名称：imagefolder
    类型：imagefolder
    配置：默认
    拆分：训练集
    参数：默认
    评估指标：
    • 名称：准确率
      类型：accuracy
      值：0.95
    • 名称：精确率
      类型：precision
      值：0.9652777777777778
    • 名称：召回率
      类型：recall
      值：0.9788732394366197
    • 名称：F1分数
      类型：f1
      值：0.972027972027972

resnet-18特征提取

该模型是基于microsoft/resnet-18在图像文件夹数据集上微调的版本。
在评估集上取得了以下结果：

• 损失：0.1485
• 准确率：0.95
• 精确率：0.9653
• 召回率：0.9789
• F1分数：0.9720
• ROC AUC：0.8505

模型描述

需补充更多信息

预期用途与限制

需补充更多信息

训练与评估数据

需补充更多信息

训练过程

训练超参数

训练过程中使用了以下超参数：

• 学习率：2e-05
• 训练批次大小：64
• 评估批次大小：64
• 随机种子：42
• 梯度累积步数：4
• 总训练批次大小：256
• 优化器：Adam（beta1=0.9，beta2=0.999，epsilon=1e-08）
• 学习率调度器类型：线性
• 学习率预热比例：0.1
• 训练轮数：50

训练结果

框架版本

• Transformers 4.24.0.dev0
• PyTorch 1.11.0+cu102
• Datasets 2.6.1
• Tokenizers 0.13.1