philgrey commited on
Commit
7702d59
1 Parent(s): df2791a

Training in progress, epoch 1

Browse files
Files changed (4) hide show
  1. README.md +3 -4
  2. config.json +4 -6
  3. model.safetensors +2 -2
  4. training_args.bin +1 -1
README.md CHANGED
@@ -6,7 +6,7 @@ tags:
6
  metrics:
7
  - accuracy
8
  model-index:
9
- - name: API_Detector_3_Distilbert
10
  results: []
11
  ---
12
 
@@ -15,15 +15,14 @@ should probably proofread and complete it, then remove this comment. -->
15
 
16
  # my_awesome_model
17
 
18
- This model is a fine-tuned version of [distilbert/distilbert-base-uncased](https://huggingface.co/distilbert/distilbert-base-uncased) on Custom dataset (3 classes).
19
  It achieves the following results on the evaluation set:
20
  - Loss: 0.6536
21
  - Accuracy: 0.9437
22
 
23
  ## Model description
24
 
25
- Due to lack of training data, this model is not recognizing difference between "order" and "orderline" in sentences. So classify "ORDERLINE" to "MAOORDER" when word "orderline" is in sentence.
26
- This can be solved by one more 'if' condition in further script and this is more efficient way than more training data and its train.
27
 
28
  ## Intended uses & limitations
29
 
 
6
  metrics:
7
  - accuracy
8
  model-index:
9
+ - name: my_awesome_model
10
  results: []
11
  ---
12
 
 
15
 
16
  # my_awesome_model
17
 
18
+ This model is a fine-tuned version of [distilbert/distilbert-base-uncased](https://huggingface.co/distilbert/distilbert-base-uncased) on the None dataset.
19
  It achieves the following results on the evaluation set:
20
  - Loss: 0.6536
21
  - Accuracy: 0.9437
22
 
23
  ## Model description
24
 
25
+ More information needed
 
26
 
27
  ## Intended uses & limitations
28
 
config.json CHANGED
@@ -9,15 +9,13 @@
9
  "dropout": 0.1,
10
  "hidden_dim": 3072,
11
  "id2label": {
12
- "0": "itemtype",
13
- "1": "orderline",
14
- "2": "maoorder"
15
  },
16
  "initializer_range": 0.02,
17
  "label2id": {
18
- "itemtype": 0,
19
- "maoorder": 2,
20
- "orderline": 1
21
  },
22
  "max_position_embeddings": 512,
23
  "model_type": "distilbert",
 
9
  "dropout": 0.1,
10
  "hidden_dim": 3072,
11
  "id2label": {
12
+ "0": "NEGATIVE",
13
+ "1": "POSITIVE"
 
14
  },
15
  "initializer_range": 0.02,
16
  "label2id": {
17
+ "NEGATIVE": 0,
18
+ "POSITIVE": 1
 
19
  },
20
  "max_position_embeddings": 512,
21
  "model_type": "distilbert",
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:faac19276a0a7bf446ab16880ce2cb26446b23fda5ac66b4b0e18960e7b9ed84
3
- size 267835644
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:36b5db81fb3ce4dfcc832d46507d0f441996ab33fe45a6ce86b9003479788195
3
+ size 267832560
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:4b355128ccb8ad8b787197d78949d6dfbe23d98b68f821382382e161e7f2906b
3
  size 4984
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0441cc05398a3596c269b736b6fde357e1a87baff07a5da7c4d1d237b70acde2
3
  size 4984