📅  最后修改于: 2022-03-11 14:45:16.275000             🧑  作者: Mango
from sklearn.utils import class_weight
class_weights = class_weight.compute_class_weight('balanced',
np.unique(y_train),
y_train)
model.fit(X_train, y_train, class_weight=class_weights)