| { | |
| "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 | |
| } | |
| } |