许可证：apache-2.0
标签：

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

模型索引：

• 名称：vit-base-patch16-224-in21k_Simpsons_Family_Members
  结果：
  • 任务：
    名称：图像分类
    类型：image-classification
    数据集：
    名称：imagefolder
    类型：imagefolder
    配置：默认
    分割：训练集
    参数：默认
    评估指标：
    • 名称：准确率
      类型：accuracy
      值：0.9529702970297029

语言：

• en

管道标签：

• image-classification

vit-base-patch16-224-in21k_Simpsons_Family_Members

该模型是基于google/vit-base-patch16-224-in21k微调后的版本。

在评估集上取得了以下结果：

• 损失：0.2431
• 准确率：0.9530
• F1值
  • 加权：0.9522
  • 微平均：0.9530
  • 宏平均：0.9521
• 召回率
  • 加权：0.9530
  • 微平均：0.9530
  • 宏平均：0.9531
• 精确率
  • 加权：0.9605
  • 微平均：0.9530
  • 宏平均：0.9601

模型描述

这是一个用于分类《辛普森一家》成员的多类别图像分类模型。

关于模型创建过程的更多信息，请查看以下链接：
https://github.com/DunnBC22/Vision_Audio_and_Multimodal_Projects/blob/main/Computer%20Vision/Image%20Classification/Multiclass%20Classification/Simpsons%20Family%20Images/Simpsons_family_with_hf_ViT.ipynb

预期用途与限制

该模型旨在展示我运用技术解决复杂问题的能力。

训练与评估数据

数据集来源：https://www.kaggle.com/datasets/williamu32/dataset-bart-or-homer

数据集中的示例图像：

训练过程

训练超参数

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

• 学习率：0.0002
• 训练批次大小：16
• 评估批次大小：8
• 随机种子：42
• 优化器：Adam（beta1=0.9，beta2=0.999，epsilon=1e-08）
• 学习率调度器类型：线性
• 训练轮次：3

训练结果

框架版本

• Transformers 4.22.2
• PyTorch 1.12.1
• Datasets 2.5.2
• Tokenizers 0.12.1