laureBe commited on
Commit
96dd54b
·
verified ·
1 Parent(s): 4096ab5

Update tasks/text.py

Browse files
Files changed (1) hide show
  1. 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
- 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
 
 
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