{ "d": 4096, "hidden": 512, "features": "/home/diffusedit/features/full/train", "features_after": "/home/diffusedit/features/full/after_train", "pairs": "/home/diffusedit/data/split/links/train/pairs.jsonl", "layer": 32, "seed": 42, "best_val_auroc": 0.9451081101225768, "tau_c": 0.35, "val_f1": 0.7651331719128329, "history": [ { "epoch": 1, "train_loss": 1.605, "val_loss": 0.892, "val_auroc": 0.8385, "seconds": 0.6 }, { "epoch": 2, "train_loss": 0.719, "val_loss": 0.7097, "val_auroc": 0.8862, "seconds": 0.3 }, { "epoch": 3, "train_loss": 0.5492, "val_loss": 0.6318, "val_auroc": 0.9139, "seconds": 0.2 }, { "epoch": 4, "train_loss": 0.407, "val_loss": 0.5501, "val_auroc": 0.9295, "seconds": 0.2 }, { "epoch": 5, "train_loss": 0.3096, "val_loss": 0.5429, "val_auroc": 0.934, "seconds": 0.2 }, { "epoch": 6, "train_loss": 0.2175, "val_loss": 0.5164, "val_auroc": 0.94, "seconds": 0.2 }, { "epoch": 7, "train_loss": 0.1577, "val_loss": 0.6775, "val_auroc": 0.9395, "seconds": 0.2 }, { "epoch": 8, "train_loss": 0.1133, "val_loss": 0.6663, "val_auroc": 0.9369, "seconds": 0.2 }, { "epoch": 9, "train_loss": 0.0863, "val_loss": 0.7139, "val_auroc": 0.9408, "seconds": 0.2 }, { "epoch": 10, "train_loss": 0.0585, "val_loss": 0.7103, "val_auroc": 0.9451, "seconds": 0.2 }, { "epoch": 11, "train_loss": 0.0422, "val_loss": 0.8489, "val_auroc": 0.9381, "seconds": 0.2 }, { "epoch": 12, "train_loss": 0.0293, "val_loss": 0.7948, "val_auroc": 0.9427, "seconds": 0.2 }, { "epoch": 13, "train_loss": 0.0217, "val_loss": 0.8185, "val_auroc": 0.9409, "seconds": 0.2 } ], "test": { "auroc": 0.9314635445435178, "precision": 0.6846153846153846, "recall": 0.7672413793103449, "f1": 0.7235772357723578, "pairs": 1106, "positives": 232 } }