Selina233 / mlpda-project

使用IMDB数据集,进行机器学习模型(传统、transformer)的训练和评估

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

mlpda-project

使用IMDB数据集,进行机器学习模型(传统、transformer)的训练和评估

实验数据

model train test(大测试集)
transformer500 0.883(小训练集) 0.7522
transformer_full_70 0.77796(大训练集) 0.76676
lstm5000 0.812(小训练集) 0.63704

*注:“小训练集”指的是train_data[12000:13000],“大训练集”指的是全部的train_data。

About

使用IMDB数据集,进行机器学习模型(传统、transformer)的训练和评估


Languages

Language:Jupyter Notebook 91.0%Language:Python 9.0%