Views
No views yet
bert-base-uncased on the emotions datasettorchpip install transformers datasets scikit-learn wandb seaborn python-dotenv.env.example to .env and set an API key from wandbexplainableai.py file.pytorch_model.bin) is a based on the bert-base-uncased and already trained on the emotions dataset.
To re-produce the training run finetune-emotions.py. You can change the base model, or the dataset by changing that file's code.example.pybert-base-uncased with the emotions dataset. However, you can change parameters and re-fine-tune the model by running finetune-emotions.py.