在为泰坦尼克号问题实施 XGboost 时出现此错误
xgb.iter.update(bst$handle, dtrain, iteration - 1, obj) 中的错误:
[03:26:03] amalgamation/../src/objective/multiclass_obj.cc:75:检查失败:label_error >= 0 && label_error < nclass SoftmaxMultiClassObj: 标签必须在 [0, num_class), num_class=2 但在标签中发现 2。
以下是我的代码:
#Parameter ie no of class
nc <- length(unique(train_label))
nc
xgb_params <- list("objective"="multi:softprob",
"eval_metric"="mlogloss",
"num_class"=nc)
watchlist <- list(train=train_matix,test=test_matix)
#XGB Model
bst_model <- xgb.train(params = xgb_params,data = train_matix, nrounds = 100,watchlist = watchlist)
Run Code Online (Sandbox Code Playgroud)
我该如何解决这个问题?