catboost案例

from catboost import CatBoostClassifier

# 数据集
cat_features = [0, 1]  # 类别特征下标
train_data = [["a", "b", 1, 4, 5, 6], 
              ["a", "b", 4, 5, 6, 7],
              ["c", "d", 30, 40, 50, 60]]
train_labels = [1, 1, -1]
eval_data = [["a", "b", 2, 4, 6, 8], 
             ["a", "d", 1, 4, 50, 60]]

# 定义模型
model = CatBoostClassifier(iterations=2, learning_rate=1, depth=2)

# 训练
model.fit(train_data, train_labels, cat_features)

# 预测类别
preds_class = model.predict(eval_data)

# 预测每个类别的可能性
preds_proba = model.predict_proba(eval_data)

# 预测Raw formula value
preds_raw = model.predict(eval_data, prediction_type='RawFormulaVal')

print(preds_class)
print(preds_proba)
print(preds_raw)
'''
[1 1]
[