Training in progress epoch 2

README.md

@@ -15,13 +15,13 @@ probably proofread and complete it, then remove this comment. -->
 
 This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Train Loss: 0.0295
-- Validation Loss: 0.0851
-- Train Precision: 0.8743
-- Train Recall: 0.8835
-- Train F1: 0.8789
-- Train Accuracy: 0.9748
-- Epoch: 1
+- Train Loss: 0.0163
+- Validation Loss: 0.1024
+- Train Precision: 0.8763
+- Train Recall: 0.8862
+- Train F1: 0.8812
+- Train Accuracy: 0.9750
+- Epoch: 2
 
 ## Model description
 
@@ -49,6 +49,7 @@ The following hyperparameters were used during training:
 |:----------:|:---------------:|:---------------:|:------------:|:--------:|:--------------:|:-----:|
 | 0.0709     | 0.0710          | 0.8563          | 0.8875       | 0.8716   | 0.9735         | 0     |
 | 0.0295     | 0.0851          | 0.8743          | 0.8835       | 0.8789   | 0.9748         | 1     |
+| 0.0163     | 0.1024          | 0.8763          | 0.8862       | 0.8812   | 0.9750         | 2     |
 
 
 ### Framework versions

logs/train/events.out.tfevents.1701601690.snek.69265.0.v2

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:5b05770922e48f35a531da13fd7d776a6f91c3bc48903f378f4164722bb6925d
-size 1606986
+oid sha256:155f12c153d08319d8cd2e230cd200761e82942848c5506542a05293e61c7a79
+size 1607335

logs/validation/events.out.tfevents.1701603726.snek.69265.1.v2

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:0cceac4e2a6db23c5603e54e6929a0c8246fa1ef73f2627df63c13ad9453caa3
-size 390
+oid sha256:a6f9e04ec9a457e4db65c55aef005ffa22f50df2638a698952ac238a31e6630f
+size 547

metrics_by_entity.txt

@@ -2,3 +2,4 @@
 {"ANIM": {"precision": 0.6876374835019797, "recall": 0.6928191489361702, "f1": 0.6902185913005079, "number": "4512"}, "BIO": {"precision": 0.4157303370786517, "recall": 0.8043478260869565, "f1": 0.5481481481481483, "number": "92"}, "CEL": {"precision": 0.9009009009009009, "recall": 0.8264462809917356, "f1": 0.8620689655172413, "number": "242"}, "DIS": {"precision": 0.7545427464998511, "recall": 0.6964531207038768, "f1": 0.7243351444094939, "number": "7274"}, "EVE": {"precision": 0.9406528189910979, "recall": 0.9161849710982659, "f1": 0.9282576866764275, "number": "692"}, "FOOD": {"precision": 0.6681350954478708, "recall": 0.5287623474723998, "f1": 0.5903340901719104, "number": "6884"}, "INST": {"precision": 0.6363636363636364, "recall": 0.603448275862069, "f1": 0.6194690265486726, "number": "116"}, "LOC": {"precision": 0.9646216239537493, "recall": 0.9754799301919721, "f1": 0.9700203913401884, "number": "22920"}, "MEDIA": {"precision": 0.9566446748350612, "recall": 0.9111310592459605, "f1": 0.9333333333333333, "number": "2228"}, "MYTH": {"precision": 0.9156626506024096, "recall": 0.7238095238095238, "f1": 0.8085106382978723, "number": "210"}, "ORG": {"precision": 0.9430745015739769, "recall": 0.9303830227743272, "f1": 0.936685773840542, "number": "7728"}, "PER": {"precision": 0.989819810648478, "recall": 0.9921428571428571, "f1": 0.990979972481272, "number": "19600"}, "PLANT": {"precision": 0.6755162241887905, "recall": 0.7124444444444444, "f1": 0.6934890763573436, "number": "4500"}, "TIME": {"precision": 0.8427299703264095, "recall": 0.6876513317191283, "f1": 0.7573333333333333, "number": "826"}, "VEHI": {"precision": 0.8617886178861789, "recall": 0.762589928057554, "f1": 0.8091603053435114, "number": "278"}, "overall_precision": 0.8909219597871392, "overall_recall": 0.8702978156769353, "overall_f1": 0.880489131842794, "overall_accuracy": 0.9754610707728644}
 {"ANIM": {"precision": 0.6991504247876063, "recall": 0.6201241134751773, "f1": 0.6572703782006107, "number": "4512"}, "BIO": {"precision": 0.48, "recall": 0.782608695652174, "f1": 0.5950413223140497, "number": "92"}, "CEL": {"precision": 0.8153846153846154, "recall": 0.8760330578512396, "f1": 0.844621513944223, "number": "242"}, "DIS": {"precision": 0.6289389067524116, "recall": 0.8067088259554578, "f1": 0.7068176343049868, "number": "7274"}, "EVE": {"precision": 0.898876404494382, "recall": 0.9248554913294798, "f1": 0.9116809116809117, "number": "692"}, "FOOD": {"precision": 0.6247739602169982, "recall": 0.6022661243463103, "f1": 0.6133136094674556, "number": "6884"}, "INST": {"precision": 0.4673913043478261, "recall": 0.7413793103448276, "f1": 0.5733333333333334, "number": "116"}, "LOC": {"precision": 0.9673544006267956, "recall": 0.9696335078534032, "f1": 0.9684926134135181, "number": "22920"}, "MEDIA": {"precision": 0.9112271540469974, "recall": 0.9398563734290843, "f1": 0.9253203711886876, "number": "2228"}, "MYTH": {"precision": 0.9418604651162791, "recall": 0.7714285714285715, "f1": 0.8481675392670157, "number": "210"}, "ORG": {"precision": 0.9244705281959683, "recall": 0.9376293995859213, "f1": 0.931003469099319, "number": "7728"}, "PER": {"precision": 0.9886988393402566, "recall": 0.9909183673469387, "f1": 0.9898073590867394, "number": "19600"}, "PLANT": {"precision": 0.6225212464589235, "recall": 0.7813333333333333, "f1": 0.6929444225463145, "number": "4500"}, "TIME": {"precision": 0.7439024390243902, "recall": 0.738498789346247, "f1": 0.7411907654921019, "number": "826"}, "VEHI": {"precision": 0.7857142857142857, "recall": 0.7913669064748201, "f1": 0.7885304659498209, "number": "278"}, "overall_precision": 0.8563028116815734, "overall_recall": 0.8875060817904791, "overall_f1": 0.8716252750707325, "overall_accuracy": 0.9734982332155477}
 {"ANIM": {"precision": 0.7010395010395011, "recall": 0.7473404255319149, "f1": 0.7234499034541944, "number": "4512"}, "BIO": {"precision": 0.4222222222222222, "recall": 0.8260869565217391, "f1": 0.5588235294117647, "number": "92"}, "CEL": {"precision": 0.7291666666666666, "recall": 0.8677685950413223, "f1": 0.7924528301886793, "number": "242"}, "DIS": {"precision": 0.7310991957104558, "recall": 0.7497937860874347, "f1": 0.7403284919234424, "number": "7274"}, "EVE": {"precision": 0.8648648648648649, "recall": 0.9248554913294798, "f1": 0.8938547486033519, "number": "692"}, "FOOD": {"precision": 0.6528651871480623, "recall": 0.5726321905868681, "f1": 0.6101222720941032, "number": "6884"}, "INST": {"precision": 0.8717948717948718, "recall": 0.5862068965517241, "f1": 0.7010309278350514, "number": "116"}, "LOC": {"precision": 0.9682553563620464, "recall": 0.9661431064572426, "f1": 0.9671980781830094, "number": "22920"}, "MEDIA": {"precision": 0.9463459759481961, "recall": 0.9183123877917415, "f1": 0.932118451025057, "number": "2228"}, "MYTH": {"precision": 0.9148936170212766, "recall": 0.819047619047619, "f1": 0.864321608040201, "number": "210"}, "ORG": {"precision": 0.9060636182902585, "recall": 0.9435817805383023, "f1": 0.9244421906693712, "number": "7728"}, "PER": {"precision": 0.9903885480572597, "recall": 0.9883673469387755, "f1": 0.9893769152196119, "number": "19600"}, "PLANT": {"precision": 0.6258885147774037, "recall": 0.7435555555555555, "f1": 0.6796668697948406, "number": "4500"}, "TIME": {"precision": 0.7559808612440191, "recall": 0.7651331719128329, "f1": 0.760529482551143, "number": "826"}, "VEHI": {"precision": 0.7375, "recall": 0.8489208633093526, "f1": 0.7892976588628763, "number": "278"}, "overall_precision": 0.8742841214332776, "overall_recall": 0.883485698189547, "overall_f1": 0.8788608255957613, "overall_accuracy": 0.9747967967229146}
+{"ANIM": {"precision": 0.7304429783223374, "recall": 0.6870567375886525, "f1": 0.7080858839652809, "number": "4512"}, "BIO": {"precision": 0.33088235294117646, "recall": 0.9782608695652174, "f1": 0.49450549450549447, "number": "92"}, "CEL": {"precision": 0.7985611510791367, "recall": 0.9173553719008265, "f1": 0.8538461538461539, "number": "242"}, "DIS": {"precision": 0.7381081081081081, "recall": 0.7508935936211163, "f1": 0.7444459588387623, "number": "7274"}, "EVE": {"precision": 0.9048991354466859, "recall": 0.9075144508670521, "f1": 0.9062049062049063, "number": "692"}, "FOOD": {"precision": 0.6286635404454866, "recall": 0.6231841952353283, "f1": 0.6259118762766268, "number": "6884"}, "INST": {"precision": 0.5333333333333333, "recall": 0.5517241379310345, "f1": 0.5423728813559322, "number": "116"}, "LOC": {"precision": 0.9679153659382588, "recall": 0.9739965095986038, "f1": 0.970946416144746, "number": "22920"}, "MEDIA": {"precision": 0.9496268656716418, "recall": 0.9138240574506283, "f1": 0.9313815187557183, "number": "2228"}, "MYTH": {"precision": 0.91, "recall": 0.8666666666666667, "f1": 0.8878048780487805, "number": "210"}, "ORG": {"precision": 0.9249199231262012, "recall": 0.9341356107660456, "f1": 0.9295049249983905, "number": "7728"}, "PER": {"precision": 0.9895058583800306, "recall": 0.9910204081632653, "f1": 0.9902625541677288, "number": "19600"}, "PLANT": {"precision": 0.6376479572355861, "recall": 0.7422222222222222, "f1": 0.6859724789484494, "number": "4500"}, "TIME": {"precision": 0.7663551401869159, "recall": 0.7941888619854721, "f1": 0.7800237812128418, "number": "826"}, "VEHI": {"precision": 0.8181818181818182, "recall": 0.6474820143884892, "f1": 0.7228915662650601, "number": "278"}, "overall_precision": 0.8762865715479371, "overall_recall": 0.8862385086169369, "overall_f1": 0.8812344437300673, "overall_accuracy": 0.9750378267167332}

tf_model.h5

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:25d93261c410b5f8f0fe20395e61d8a175202107c05e990b33d00bbfcc0db4e4
+oid sha256:df09a97559d027620f58b276ce451883474a4af4457841ea430f6fb094d3638f
 size 265674000