前提・実現したいこと
1.エラーを回避させて動作させる。
listではないのにエラーが発生します。num_classに問題があるのでしょうか?
有識者の方々、ご助力お願いいたします。
発生している問題・エラーメッセージ
TypeError Traceback (most recent call last)
<ipython-input-126-ccfa323308ad> in <module>
17
18 num_round = 10
---> 19 bst = xgb.train(param_grid, dtrain, num_round)
20
21
~\Anaconda3\lib\site-packages\xgboost\training.py in train(params, dtrain, num_boost_round, evals, obj, feval, maximize, early_stopping_rounds, evals_result, verbose_eval, xgb_model, callbacks, learning_rates)
214 evals=evals,
215 obj=obj, feval=feval,
--> 216 xgb_model=xgb_model, callbacks=callbacks)
217
218
~\Anaconda3\lib\site-packages\xgboost\training.py in _train_internal(params, dtrain, num_boost_round, evals, obj, feval, xgb_model, callbacks)
46 nboost //= num_parallel_tree
47 if 'num_class' in _params:
---> 48 nboost //= _params['num_class']
49
50 # Distributed code: Load the checkpoint from rabit.
TypeError: unsupported operand type(s) for //=: 'int' and 'list'
該当のソースコード
dtrain = xgb.DMatrix(x_train, label=y_train)
dtest = xgb.DMatrix(x_test, label=y_test)
param_grid = {
'max_depth': [0,1,3],
'gamma':[0,0.2,0.03],
'booster':['gbtree'],
'objective':['multi:softmax'],
'num_class':[50],
}
num_round = 10
bst = xgb.train(param_grid, dtrain, num_round)
xgb.plot_importance(bst)
回答1件
あなたの回答
tips
プレビュー