query_classifier_svm / config.json
yyyyyy23's picture
Create config.json
0e94000 verified
raw
history blame
No virus
754 Bytes
{
"model_type": "SVC",
"framework": "scikit-learn",
"task": "text-classification",
"labels": ["math", "web question", "greeting", "comparative question", "sherpa question"],
"id2label": {
"0": "math",
"1": "web question",
"2": "greeting",
"3": "comparative question",
"4": "sherpa question"
},
"label2id": {
"math": 0,
"web question": 1,
"greeting": 2,
"comparative question": 3,
"sherpa question": 4
},
"model_parameters": {
"C": 1.0,
"kernel": "rbf",
"degree": 3,
"gamma": "scale"
},
"training_parameters": {
"train_test_split_ratio": 0.8,
"random_state": 42,
"data_preprocessing": {
"text_vectorization": "TF-IDF",
"max_features": 10000
}
}
}