Getting this error while implementing XGboost for the titanic problem
Error in xgb.iter.update(bst$handle, dtrain, iteration - 1, obj) :
[03:26:03] amalgamation/../src/objective/multiclass_obj.cc:75: Check failed: label_error >= 0 && label_error < nclass SoftmaxMultiClassObj: label must be in [0, num_class), num_class=2 but found 2 in label.
Following is my code:
#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)
How can I resolve this?