【源码】基于PyTorch框架的多模态情感分析系统

项目简介

本项目是基于PyTorch框架开发的多模态情感分析系统，能够结合文本和图像数据，对配对数据进行情感标签预测，支持积极、中性和消极三分类任务。

项目的主要特性和功能

• 采用多模态模型，融合文本与图像数据开展情感分析。
• 文本模型借助预训练的BERT模型提取文本特征。
• 图像模型使用简单的神经网络模型提取图像特征。
• 具备数据预处理、训练、验证和测试等数据处理功能。
• 可通过命令行参数灵活配置训练、测试和预测结果保存。

安装使用步骤

前提条件

用户需已安装Python环境，且具备Python、PyTorch、Transformers等库的基本知识。

安装依赖

在命令行运行以下命令安装项目所需依赖： shell pip install -r requirements.txt

使用流程

1. 下载项目源码文件。
2. 依据config.py中的说明配置项目参数。
3. 运行模型训练和评估：
4. 训练模型： shell python main.py --output_dir output/ --train_filepath dataset/train.json --dev_filepath dataset/dev.json --test_filepath dataset/test.json --do_train
5. 测试模型并保存预测结果： shell python main.py --output_dir output/ --train_filepath dataset/train.json --dev_filepath dataset/dev.json --test_filepath dataset/test.json --do_test --store_preds
6. 查看实验结果和预测结果。

下载地址

点击下载 【提取码: 4003】【解压密码: www.makuang.net】