В настоящее время мы используем версию Python библиотеки catboost v0.14, но мы думаем о переходе с библиотеки Python на библиотеку Java, потому что из нашего кода мы в настоящее время вызываем код Python.поэтому не могли бы вы сказать нам, какая версия библиотеки Java соответствует этой библиотеке Python.
Я пытался https://mvnrepository.com/artifact/ai.catboost/catboost-prediction/0.2.5, но оценка совпадает только в 28% тестовых данных.
String bestModelFilePath = args.length > 0 && args[0] != null ? args[0]:
"best_model.cbm";
String testDataFilePath = args.length > 1 && args[1] != null ? args[1]:
"test_data.csv";
String outputFilePath = args.length > 2 && args[2] != null ? args[2]:
"output_file.txt";
CatBoostModel model = CatBoostModel.loadModel(bestModelFilePath);
CatBoostPredictions prediction = model.predict(floatFeatures,
catFeatures);
File file = new File(outputFilePath);
if(file.exists()) {
file.delete();
}
file.createNewFile();
FileWriter writer = new FileWriter(file);
writer.write(sigmoid(prediction.get(0, 0))+"\n");
оценка из библиотеки Python и Java должна совпадать, но они не совпадают