{ "epoch": 4, "val_metrics": { "accuracy": 0.6023391812865497, "macro_f1": 0.5676855529936099, "classification_report": { "G": { "precision": 1.0, "recall": 0.4, "f1-score": 0.5714285714285714, "support": 10.0 }, "PG": { "precision": 0.5, "recall": 0.34782608695652173, "f1-score": 0.41025641025641024, "support": 23.0 }, "PG-13": { "precision": 0.43478260869565216, "recall": 0.23255813953488372, "f1-score": 0.30303030303030304, "support": 43.0 }, "R": { "precision": 0.6229508196721312, "recall": 0.8539325842696629, "f1-score": 0.7203791469194313, "support": 89.0 }, "NC-17": { "precision": 0.8333333333333334, "recall": 0.8333333333333334, "f1-score": 0.8333333333333334, "support": 6.0 }, "accuracy": 0.6023391812865497, "macro avg": { "precision": 0.6782133523402234, "recall": 0.5335300288188803, "f1-score": 0.5676855529936099, "support": 171.0 }, "weighted avg": { "precision": 0.5885279247060393, "recall": 0.6023391812865497, "f1-score": 0.5689721067620793, "support": 171.0 } }, "confusion_matrix": [ [ 4, 2, 1, 3, 0 ], [ 0, 8, 5, 10, 0 ], [ 0, 1, 10, 32, 0 ], [ 0, 5, 7, 76, 1 ], [ 0, 0, 0, 1, 5 ] ] }, "train_loss": 0.039685951293562324 }