Update tasks/text.py
Browse files- tasks/text.py +2 -2
tasks/text.py
CHANGED
@@ -88,8 +88,8 @@ async def evaluate_text(request: TextEvaluationRequest):
|
|
88 |
#Parameters: {'colsample_bytree': 0.7498850106268238, 'gamma': 0.3690168082131852, 'learning_rate': 0.054839600377537934, 'max_depth': 5, 'n_estimators': 125, 'subsample': 0.6272998821416366}
|
89 |
|
90 |
#xgb_model = xgb.XGBRegressor(max_depth=5, objective='multi:softprob',
|
91 |
-
|
92 |
-
|
93 |
#xgb_model.fit(tfidf_train, y_train)
|
94 |
#y_pred = xgb_model.predict(tfidf_train)
|
95 |
|
|
|
88 |
#Parameters: {'colsample_bytree': 0.7498850106268238, 'gamma': 0.3690168082131852, 'learning_rate': 0.054839600377537934, 'max_depth': 5, 'n_estimators': 125, 'subsample': 0.6272998821416366}
|
89 |
|
90 |
#xgb_model = xgb.XGBRegressor(max_depth=5, objective='multi:softprob',
|
91 |
+
# n_estimators=125, num_class=8, colsample_bytree=0.7498850106268238,gamma=0.3690168082131852,
|
92 |
+
# learning_rate=0.054839600377537934, subsample=0.6272998821416366)
|
93 |
#xgb_model.fit(tfidf_train, y_train)
|
94 |
#y_pred = xgb_model.predict(tfidf_train)
|
95 |
|