Поддерживает мультиклассовую классификацию. Ниже приведен код:
param = {
'max_depth': 3, # the maximum depth of each tree
'eta': 0.3, # the training step for each iteration
'silent': 1, # logging mode - quiet
'objective': 'multi:softprob', # error evaluation for multiclass training
'num_class': 3} # the number of classes that exist in this dataset
Вы можете использовать num_class
в качестве параметра в Python для классификации нескольких классов.