Hugging Face's logo

PIBotBCCh
/
pibot-intent-router

Text Classification
PyTorch
Safetensors
sentence-transformers
Spanish
multitask-learning
spanish
intent-classification
Model card Files
xet
 Community
smenaaliaga commited on
Commit
114862d
·
verified ·
1 Parent(s): b2df0c9

Upload multitask artifacts

Browse files
Files changed (7) hide show
  1. README.md +1 -1
  2. encoder/adapter_config.json +2 -2
  3. encoder/adapter_model.safetensors +1 -1
  4. encoder/lora_weights/adapter_config.json +2 -2
  5. encoder/lora_weights/adapter_model.safetensors +1 -1
  6. heads.pt +1 -1
  7. train_config.json +2 -2
README.md CHANGED
@@ -88,7 +88,7 @@ Salida esperada (ejemplo):
88
  Con este proyecto clonado, también puedes probar:
89
 
90
  ```bash
91
- python -m src.main test --artifact-dir artifacts --text "quiero pagar mi factura" --device cpu
92
  ```
93
 
94
  ## Notas
 
88
  Con este proyecto clonado, también puedes probar:
89
 
90
  ```bash
91
+ python -m src.main test --artifact-dir models/artifacts --text "quiero pagar mi factura" --device cpu
92
  ```
93
 
94
  ## Notas
encoder/adapter_config.json CHANGED
@@ -32,8 +32,8 @@
32
  "rank_pattern": {},
33
  "revision": null,
34
  "target_modules": [
35
- "value",
36
- "query"
37
  ],
38
  "target_parameters": null,
39
  "task_type": null,
 
32
  "rank_pattern": {},
33
  "revision": null,
34
  "target_modules": [
35
+ "query",
36
+ "value"
37
  ],
38
  "target_parameters": null,
39
  "task_type": null,
encoder/adapter_model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:ac132a9d5f67a9c390e6643385c4fad7a4eca489e6513353d8d252bf58fb1216
3
  size 298128
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:678d998e356d8d48b3ceb86189109a347f7f29e1f184b15c87d8ba9a4618275e
3
  size 298128
encoder/lora_weights/adapter_config.json CHANGED
@@ -32,8 +32,8 @@
32
  "rank_pattern": {},
33
  "revision": null,
34
  "target_modules": [
35
- "value",
36
- "query"
37
  ],
38
  "target_parameters": null,
39
  "task_type": null,
 
32
  "rank_pattern": {},
33
  "revision": null,
34
  "target_modules": [
35
+ "query",
36
+ "value"
37
  ],
38
  "target_parameters": null,
39
  "task_type": null,
encoder/lora_weights/adapter_model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:ac132a9d5f67a9c390e6643385c4fad7a4eca489e6513353d8d252bf58fb1216
3
  size 298128
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:678d998e356d8d48b3ceb86189109a347f7f29e1f184b15c87d8ba9a4618275e
3
  size 298128
heads.pt CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:c9ad0ca8d96ccb7a508732d736cb14acd651ff580c952a3f48b7fc5209476225
3
  size 15421
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c1661184ddba090e5398a41599bdb95de078aefc72028e3123468e9993273b6b
3
  size 15421
train_config.json CHANGED
@@ -2,7 +2,7 @@
2
  "model_name": "sentence-transformers/all-MiniLM-L6-v2",
3
  "max_length": 128,
4
  "epochs": 20,
5
- "batch_size": 16,
6
  "lr_encoder": 2e-05,
7
  "lr_heads": 0.001,
8
  "weight_decay": 0.01,
@@ -10,7 +10,7 @@
10
  "task_weight_intent": 1.0,
11
  "task_weight_context": 1.0,
12
  "device": "cuda",
13
- "output_dir": "artifacts",
14
  "seed": 42,
15
  "use_lora": true,
16
  "lora_r": 8,
 
2
  "model_name": "sentence-transformers/all-MiniLM-L6-v2",
3
  "max_length": 128,
4
  "epochs": 20,
5
+ "batch_size": 10,
6
  "lr_encoder": 2e-05,
7
  "lr_heads": 0.001,
8
  "weight_decay": 0.01,
 
10
  "task_weight_intent": 1.0,
11
  "task_weight_context": 1.0,
12
  "device": "cuda",
13
+ "output_dir": "models\\artifacts",
14
  "seed": 42,
15
  "use_lora": true,
16
  "lora_r": 8,