import numpy as np
from catboost import CatBoostClassifier, Pool
# initialize data
train_data = Pool("./data/ki_190625_Final280_0.csv", column_description="./data/
train2.cd",delimiter=',')
test_data = Pool("./data/ki_93k_Final280_0.csv", column_description="./data/
train3.cd",delimiter=',')
model = CatBoostClassifier(learning_rate=0.03,
custom_metric=['Accuracy'])
# train the model
model.fit(train_data)
# make the prediction using the resulting model
preds_class = model.predict(test_data)
preds_proba = model.predict_proba(test_data)
print("class = ", preds_class)
print("proba = ", preds_proba)