Skip to content
Snippets Groups Projects
bert_base_rs42_huggingfaceimdb_mmodel.txt 823 B
Newer Older
kulcsar's avatar
kulcsar committed
Namespace(architecture='bert-base-uncased', model_type='one', tmix=True, mixlayer=7, imdb=True, mlp=False, train_dataset='../datasets/data_splits/imdb_train2.txt', test_dataset='../datasets/data_splits/imdb_test2.txt', tokenizer='swp', tcontext=False, vcontext=False, masking=False, max_length=256, sample_imdb=False, sample_size=None, train_loop='swp', epochs=20, learning_rate=1e-05, second_learning_rate=0.001, random_seed=42, batch_size=4, gradient_accumulation_steps=1, mix_up=False, threshold=0.05, lambda_value=0.1, mixepoch=None, test_batch_size=24, save_directory='./Results/bert_base_rs42_huggingfaceimdb_mmodel.txt'){'accuracy': 0.7822, 'f1': 0.7561467150342602, 'precision': 0.8588869671380608, 'recall': 0.67536}{'accuracy': 0.8774, 'f1': 0.8656938784452918, 'precision': 0.9570778025385137, 'recall': 0.79024}