yyyyyy23 commited on
Commit
0e94000
1 Parent(s): 529e6ac

Create config.json

Browse files
Files changed (1) hide show
  1. config.json +34 -0
config.json ADDED
@@ -0,0 +1,34 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_type": "SVC",
3
+ "framework": "scikit-learn",
4
+ "task": "text-classification",
5
+ "labels": ["math", "web question", "greeting", "comparative question", "sherpa question"],
6
+ "id2label": {
7
+ "0": "math",
8
+ "1": "web question",
9
+ "2": "greeting",
10
+ "3": "comparative question",
11
+ "4": "sherpa question"
12
+ },
13
+ "label2id": {
14
+ "math": 0,
15
+ "web question": 1,
16
+ "greeting": 2,
17
+ "comparative question": 3,
18
+ "sherpa question": 4
19
+ },
20
+ "model_parameters": {
21
+ "C": 1.0,
22
+ "kernel": "rbf",
23
+ "degree": 3,
24
+ "gamma": "scale"
25
+ },
26
+ "training_parameters": {
27
+ "train_test_split_ratio": 0.8,
28
+ "random_state": 42,
29
+ "data_preprocessing": {
30
+ "text_vectorization": "TF-IDF",
31
+ "max_features": 10000
32
+ }
33
+ }
34
+ }