zengchen/bert-sentiment-classifier

1---2license: mit3datasets:4- imdb5tags:6- pytorch7---8对bert-base-uncased模型进行微调的情感分析，数据集采用IMDB。9 10具体参数如下：11```python12training_args = TrainingArguments(13    output_dir='./results/trainer',14    num_train_epochs=3,15    per_device_train_batch_size=8,16    per_device_eval_batch_size=8,17    warmup_steps=500,18    weight_decay=0.01,19    logging_dir='./logs',20    logging_steps=10,21    load_best_model_at_end=True,22    save_strategy='epoch',23    evaluation_strategy='epoch'24)25```26 27代码仓库：https://github.com/zengchen233/sentiment-classfier