dandankim/token_classifier2

README.md

@@ -21,11 +21,11 @@ should 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:
-- Loss: 0.0999
-- Precision: 0.9466
-- Recall: 0.9185
-- F1: 0.9323
-- Accuracy: 0.9733
+- Loss: 0.0691
+- Precision: 0.9524
+- Recall: 0.9846
+- F1: 0.9682
+- Accuracy: 0.9818
 
 ## Model description
 
@@ -56,11 +56,11 @@ The following hyperparameters were used during training:
 
 | Training Loss | Epoch | Step | Validation Loss | Precision | Recall | F1     | Accuracy |
 |:-------------:|:-----:|:----:|:---------------:|:---------:|:------:|:------:|:--------:|
-| 0.6612        | 1.0   | 28   | 0.2069          | 0.9478    | 0.9407 | 0.9442 | 0.9763   |
-| 0.1012        | 2.0   | 56   | 0.1051          | 0.9692    | 0.9333 | 0.9509 | 0.9792   |
-| 0.0482        | 3.0   | 84   | 0.0985          | 0.9627    | 0.9556 | 0.9591 | 0.9822   |
-| 0.0213        | 4.0   | 112  | 0.0992          | 0.9624    | 0.9481 | 0.9552 | 0.9792   |
-| 0.0172        | 5.0   | 140  | 0.0999          | 0.9466    | 0.9185 | 0.9323 | 0.9733   |
+| 0.0542        | 1.0   | 141  | 0.0766          | 0.9508    | 0.9815 | 0.9659 | 0.9807   |
+| 0.0776        | 2.0   | 282  | 0.0550          | 0.9595    | 0.9846 | 0.9719 | 0.9853   |
+| 0.063         | 3.0   | 423  | 0.0837          | 0.9481    | 0.9831 | 0.9653 | 0.9795   |
+| 0.0378        | 4.0   | 564  | 0.0890          | 0.9467    | 0.9846 | 0.9653 | 0.9795   |
+| 0.0588        | 5.0   | 705  | 0.0691          | 0.9524    | 0.9846 | 0.9682 | 0.9818   |
 
 
 ### Framework versions

config.json

@@ -17,8 +17,12 @@
     "5": "B-QUOTE_TOKEN",
     "6": "B-STOPLOSS",
     "7": "B-TARGET_SELL_PRICE",
-    "8": "I-ENTRY_PRICE",
-    "9": "O"
+    "8": "I-ACTION",
+    "9": "I-BASE_TOKEN",
+    "10": "I-ENTRY_PRICE",
+    "11": "I-LEVERAGE",
+    "12": "I-TARGET_SELL_PRICE",
+    "13": "O"
   },
   "initializer_range": 0.02,
   "label2id": {
@@ -30,8 +34,12 @@
     "B-QUOTE_TOKEN": 5,
     "B-STOPLOSS": 6,
     "B-TARGET_SELL_PRICE": 7,
-    "I-ENTRY_PRICE": 8,
-    "O": 9
+    "I-ACTION": 8,
+    "I-BASE_TOKEN": 9,
+    "I-ENTRY_PRICE": 10,
+    "I-LEVERAGE": 11,
+    "I-TARGET_SELL_PRICE": 12,
+    "O": 13
   },
   "max_position_embeddings": 512,
   "model_type": "distilbert",

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:2ccd1c29144eefd3e65158d0bf3752311ac298b9aaee6c63cf50ae9947a65ed1
-size 265494624
+oid sha256:4340fc2106c7e6ea005e1abb7aa3a8acb8a463ee69adb0ede9d6a59a8b7bb9a9
+size 265506928

runs/Feb12_06-38-03_5754239aeac9/events.out.tfevents.1739342285.5754239aeac9.303.4

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4723a2a7a9f40a18c595edf9c397d046e896269af95d95ff1a07f37e185756d0
+size 23127

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:6170693d4b72291ae4d45cb225c2220d3eaa39600414c0bf1c69a316d4e23bd2
+oid sha256:fc23b2f6500e05987d0d9a69156e7a0e41eb81b30ef872901450636f55d94153
 size 5368