Add Cuda iniitialization in Classifier

This commit is contained in:
2024-09-18 12:13:11 +02:00
parent d0955d9369
commit baa631dd66
5 changed files with 36 additions and 14 deletions

View File

@@ -38,6 +38,7 @@ namespace bayesnet {
std::string dump_cpt() const override;
void setHyperparameters(const nlohmann::json& hyperparameters) override; //For classifiers that don't have hyperparameters
protected:
torch::Device device;
bool fitted;
unsigned int m, n; // m: number of samples, n: number of features
Network model;