Modelpublic

DrinkIcedT/roberta-large_MBTI_I_MBTI_agg_balanced_50_2e-05

sourceHugging Faceupdated 1mo agoView on Hugging Face

0likes14downloads

Model Card

roberta-largeMBTIIMBTIaggbalanced50_2e-05

This model was trained from scratch on an unknown dataset. It achieves the following results on the evaluation set:

—Loss: 6.4483
—F1: 0.6724
—Threshold: 0.66
—F1 At 05: 0.6649

Model description

More information needed

Intended uses & limitations

More information needed

Training and evaluation data

More information needed

Training procedure

Training hyperparameters

The following hyperparameters were used during training:

—learning_rate: 2e-05
—trainbatchsize: 16
—evalbatchsize: 16
—seed: 42
—distributed_type: multi-GPU
—num_devices: 4
—totaltrainbatch_size: 64
—totalevalbatch_size: 64
—optimizer: Use OptimizerNames.ADAMWTORCH with betas=(0.9,0.999) and epsilon=1e-08 and optimizerargs=No additional optimizer arguments
—lrschedulertype: linear
—lrschedulerwarmup_steps: 400
—num_epochs: 5

Training results

Training Loss	Epoch	Step	Validation Loss	F1	Threshold	F1 At 05
2.6790	0.3175	200	2.5845	0.6344	0.39	0.5363
2.4516	0.6349	400	2.4236	0.6571	0.52	0.6542
2.3273	0.9524	600	2.3726	0.6723	0.51	0.6717
1.9333	1.2698	800	2.4694	0.6742	0.4600	0.6700
1.8594	1.5873	1000	2.4772	0.6874	0.35	0.6783
1.7841	1.9048	1200	2.4794	0.6846	0.38	0.6700
1.1805	2.2222	1400	3.1633	0.6807	0.35	0.6762
1.2838	2.5397	1600	2.9594	0.6715	0.38	0.6676
1.2491	2.8571	1800	3.2493	0.6755	0.4400	0.6717
0.6813	3.1746	2000	4.1686	0.6729	0.47	0.6688
0.6933	3.4921	2200	4.4958	0.6691	0.71	0.6667
0.6468	3.8095	2400	4.2346	0.6779	0.59	0.6728
0.3538	4.1270	2600	5.9187	0.6730	0.49	0.6723
0.3386	4.4444	2800	6.1454	0.6692	0.7	0.6666
0.3529	4.7619	3000	6.4483	0.6724	0.66	0.6649

Framework versions

—Transformers 5.3.0
—Pytorch 2.7.1
—Datasets 4.8.5
—Tokenizers 0.22.2