Training in progress epoch 1

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.0709
-- Validation Loss: 0.0710
-- Train Precision: 0.8563
-- Train Recall: 0.8875
-- Train F1: 0.8716
-- Train Accuracy: 0.9735
-- Epoch: 0
+- 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
 
 ## Model description
 
@@ -48,6 +48,7 @@ The following hyperparameters were used during training:
 | Train Loss | Validation Loss | Train Precision | Train Recall | Train F1 | Train Accuracy | Epoch |
 |:----------:|:---------------:|:---------------:|:------------:|:--------:|:--------------:|:-----:|
 | 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     |
 
 
 ### 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:e5513cd9960851e0335f68685068f3c5f0c7fb1a93a39afd0efc8d3feb145c42
-size 1606637
+oid sha256:5b05770922e48f35a531da13fd7d776a6f91c3bc48903f378f4164722bb6925d
+size 1606986

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

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

metrics_by_entity.txt

@@ -1,3 +1,4 @@
 {"ANIM": {"precision": 0.7339578454332553, "recall": 0.6945921985815603, "f1": 0.7137326349350945, "number": "4512"}, "BIO": {"precision": 0.5217391304347826, "recall": 0.782608695652174, "f1": 0.6260869565217392, "number": "92"}, "CEL": {"precision": 0.695364238410596, "recall": 0.8677685950413223, "f1": 0.7720588235294118, "number": "242"}, "DIS": {"precision": 0.7090471092077087, "recall": 0.7283475391806434, "f1": 0.7185677471856775, "number": "7274"}, "EVE": {"precision": 0.9156976744186046, "recall": 0.9104046242774566, "f1": 0.9130434782608694, "number": "692"}, "FOOD": {"precision": 0.6234283961974855, "recall": 0.5906449738524114, "f1": 0.6065940623601372, "number": "6884"}, "INST": {"precision": 0.5660377358490566, "recall": 0.5172413793103449, "f1": 0.5405405405405406, "number": "116"}, "LOC": {"precision": 0.9640635798203179, "recall": 0.9738219895287958, "f1": 0.9689182149678763, "number": "22920"}, "MEDIA": {"precision": 0.946927374301676, "recall": 0.9129263913824057, "f1": 0.9296160877513711, "number": "2228"}, "MYTH": {"precision": 0.8240740740740741, "recall": 0.8476190476190476, "f1": 0.8356807511737089, "number": "210"}, "ORG": {"precision": 0.9337825579888558, "recall": 0.9324534161490683, "f1": 0.9331175137584978, "number": "7728"}, "PER": {"precision": 0.9934311813609771, "recall": 0.9876530612244898, "f1": 0.9905336949291306, "number": "19600"}, "PLANT": {"precision": 0.6034673552194763, "recall": 0.7271111111111112, "f1": 0.6595444466841363, "number": "4500"}, "TIME": {"precision": 0.762402088772846, "recall": 0.7070217917675545, "f1": 0.7336683417085427, "number": "826"}, "VEHI": {"precision": 0.7950819672131147, "recall": 0.697841726618705, "f1": 0.7432950191570882, "number": "278"}, "overall_precision": 0.8724332159340868, "overall_recall": 0.878543443189675, "overall_f1": 0.8754776684040293, "overall_accuracy": 0.9740321896133372}
 {"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}

tf_model.h5

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