distilbert-imdb-sentiment / final_results.json
Hums003's picture
Upload DistilBERT IMDB sentiment model and results
66bf654 verified
{
"test_accuracy": 0.946,
"test_f1": 0.9722507708119219,
"best_val_accuracy": 0.93,
"best_epoch": 3,
"total_parameters": 66955010,
"trainable_parameters": 66955010,
"training_samples": 16000,
"test_samples": 5000,
"epochs_trained": 3,
"batch_size": 16,
"learning_rate": 2e-05,
"max_length": 512
}