Я имею дело с высокими несбалансированными данными в проблеме классификации, поэтому я использую взвешенный класс для решения этой проблемы, но Python перестает работать, когда я просто запускаю эту ячейку ноутбука Jupyter (подходящую ячейку).
Мой код:
xg = XGBClassifier(objective ='reg:logistic', colsample_bytree = 0.3, learning_rate = 0.01,
max_depth = 5, alpha = 10, n_estimators = 100)
class_weight = {0: 1,
1: 9}
xg.fit(X_train,y_train,sample_weight=class_weight)