desertsyao's picture
Upload folder using huggingface_hub
2046dde verified
Raw
History Blame Contribute Delete
4.47 kB
[
{
"epoch": 1,
"train_loss": 0.46308579461959504,
"loss": 0.36445806480944154,
"accuracy": 0.85,
"macro_f1": 0.8474214220323466,
"labels": [
"secure_communication",
"unrelated"
],
"precision": [
0.7058823529411765,
1.0
],
"recall": [
1.0,
0.765625
],
"f1": [
0.8275862068965517,
0.8672566371681416
],
"support": [
36,
64
],
"confusion_matrix": [
[
36,
0
],
[
15,
49
]
]
},
{
"epoch": 2,
"train_loss": 0.14463981395044054,
"loss": 0.1996748084295541,
"accuracy": 0.94,
"macro_f1": 0.9369482976040353,
"labels": [
"secure_communication",
"unrelated"
],
"precision": [
0.8571428571428571,
1.0
],
"recall": [
1.0,
0.90625
],
"f1": [
0.9230769230769231,
0.9508196721311475
],
"support": [
36,
64
],
"confusion_matrix": [
[
36,
0
],
[
6,
58
]
]
},
{
"epoch": 3,
"train_loss": 0.044339864522529145,
"loss": 0.19652767550200223,
"accuracy": 0.95,
"macro_f1": 0.9438895746829761,
"labels": [
"secure_communication",
"unrelated"
],
"precision": [
1.0,
0.927536231884058
],
"recall": [
0.8611111111111112,
1.0
],
"f1": [
0.9253731343283582,
0.9624060150375939
],
"support": [
36,
64
],
"confusion_matrix": [
[
31,
5
],
[
0,
64
]
]
},
{
"epoch": 4,
"train_loss": 0.03658577402238734,
"loss": 0.23377828571712597,
"accuracy": 0.94,
"macro_f1": 0.9369482976040353,
"labels": [
"secure_communication",
"unrelated"
],
"precision": [
0.8571428571428571,
1.0
],
"recall": [
1.0,
0.90625
],
"f1": [
0.9230769230769231,
0.9508196721311475
],
"support": [
36,
64
],
"confusion_matrix": [
[
36,
0
],
[
6,
58
]
]
},
{
"epoch": 5,
"train_loss": 0.014705081137556893,
"loss": 0.24023313322803005,
"accuracy": 0.95,
"macro_f1": 0.947207264280435,
"labels": [
"secure_communication",
"unrelated"
],
"precision": [
0.8780487804878049,
1.0
],
"recall": [
1.0,
0.921875
],
"f1": [
0.935064935064935,
0.959349593495935
],
"support": [
36,
64
],
"confusion_matrix": [
[
36,
0
],
[
5,
59
]
]
},
{
"epoch": 6,
"train_loss": 0.0160370896885676,
"loss": 0.21325894113164395,
"accuracy": 0.95,
"macro_f1": 0.947207264280435,
"labels": [
"secure_communication",
"unrelated"
],
"precision": [
0.8780487804878049,
1.0
],
"recall": [
1.0,
0.921875
],
"f1": [
0.935064935064935,
0.959349593495935
],
"support": [
36,
64
],
"confusion_matrix": [
[
36,
0
],
[
5,
59
]
]
},
{
"epoch": 7,
"train_loss": 0.01584269683371531,
"loss": 0.22577850915607997,
"accuracy": 0.95,
"macro_f1": 0.947207264280435,
"labels": [
"secure_communication",
"unrelated"
],
"precision": [
0.8780487804878049,
1.0
],
"recall": [
1.0,
0.921875
],
"f1": [
0.935064935064935,
0.959349593495935
],
"support": [
36,
64
],
"confusion_matrix": [
[
36,
0
],
[
5,
59
]
]
},
{
"epoch": 8,
"train_loss": 0.011211244233951827,
"loss": 0.22725245684967377,
"accuracy": 0.95,
"macro_f1": 0.947207264280435,
"labels": [
"secure_communication",
"unrelated"
],
"precision": [
0.8780487804878049,
1.0
],
"recall": [
1.0,
0.921875
],
"f1": [
0.935064935064935,
0.959349593495935
],
"support": [
36,
64
],
"confusion_matrix": [
[
36,
0
],
[
5,
59
]
]
}
]