Fix xgboost error in predict/predict_proba

This commit is contained in:
2025-04-12 17:48:23 +02:00
parent 761f57be6c
commit 830265d91b
4 changed files with 55 additions and 30 deletions

View File

@@ -49,6 +49,7 @@ namespace pywrap {
nlohmann::json hyperparameters;
void trainModel(const torch::Tensor& weights, const bayesnet::Smoothing_t smoothing = bayesnet::Smoothing_t::NONE) override {};
std::vector<std::string> notes;
bool xgboost = false;
private:
PyWrap* pyWrap;
std::string module;