dhruvk2002
/

sentiment_analysis

scikit-learn-pipeline

Inference Endpoints

Model card Files Files and versions Community

dhruvk2002 commited on Dec 25, 2023

Commit

ea75971

•

1 Parent(s): 88eaf10

Update config.json

Files changed (1) hide show

config.json +7 -10

config.json CHANGED Viewed

@@ -1,13 +1,10 @@
 {
-    "architectures": ["RandomForestClassifier"],
-    "model_type": "random_forest",
-    "vectorizer_config": {
-      "stop_words": "english"
-    },
-    "classifier_config": {
-      "n_estimators": 100,
-      "n_jobs": -1
-    },
-    "transformers_version": "0.24.0"
   }

 {
+  "model_type": "scikit-learn-pipeline",
+  "pipeline_name": "random_forest_tfidf",
+  "model_config": {
+    "vectorizer": "TfidfVectorizer",
+    "classifier": "RandomForestClassifier"
   }
+}