flux-detector-vit / training_results.json
ash12321's picture
Upload folder using huggingface_hub
feaa2d3 verified
Raw
History Blame Contribute Delete
2.54 kB
{
"detector_name": "FLUX",
"random_seed": 123,
"best_epoch": 6,
"best_val_acc": 0.998,
"training_time_seconds": 974.2553210258484,
"test_metrics": {
"accuracy": 0.9985,
"precision": 1.0,
"recall": 0.997,
"f1": 0.9984977466199298,
"auc": 0.999994,
"fpr": 0.0,
"fnr": 0.003
},
"confusion_matrix": [
[
1000,
0
],
[
3,
997
]
],
"training_history": {
"train_loss": [
0.22600403943657876,
0.2041842999458313,
0.20400824850797653,
0.20128074333071708,
0.2000999386012554,
0.199884980738163,
0.20074790892004968,
0.1999598905146122,
0.19931599202752112
],
"train_acc": [
0.9850625,
0.9976875,
0.9971875,
0.9989375,
0.9995,
0.9995625,
0.999,
0.9994375,
0.99975
],
"val_loss": [
0.21381006401682656,
0.2048076204364262,
0.20778910342663054,
0.20300078060891893,
0.20336188730739413,
0.20366992931517344,
0.2053397801660356,
0.20481586692825196,
0.20351921920738522
],
"val_acc": [
0.995,
0.997,
0.9935,
0.9975,
0.9975,
0.998,
0.9965,
0.997,
0.997
],
"val_precision": [
0.9940119760479041,
0.998995983935743,
0.9890981169474727,
0.997997997997998,
0.997002997002997,
0.998997995991984,
0.996996996996997,
0.997,
0.998995983935743
],
"val_recall": [
0.996,
0.995,
0.998,
0.997,
0.998,
0.997,
0.996,
0.997,
0.995
],
"val_f1": [
0.995004995004995,
0.996993987975952,
0.993529118964659,
0.9974987493746873,
0.9975012493753124,
0.997997997997998,
0.9964982491245623,
0.997,
0.996993987975952
],
"val_auc": [
0.999638,
0.999973,
0.999935,
0.999962,
0.999324,
0.999968,
0.999662,
0.999729,
0.9997940000000001
]
},
"config": {
"model_name": "google/vit-base-patch16-224",
"image_size": 224,
"num_classes": 2,
"batch_size": 32,
"learning_rate": 2e-05,
"num_epochs": 20,
"early_stopping_patience": 5,
"dropout_rate": 0.1,
"label_smoothing": 0.1,
"weight_decay": 0.01
},
"dataset_info": {
"train_real": 8000,
"train_fake": 8000,
"val_real": 1000,
"val_fake": 1000,
"test_real": 1000,
"test_fake": 1000
}
}