rmontanana/BayesNet
1
0
Fork 0
Code Issues 2 Pull Requests 1 Actions Packages Projects Releases 6 Wiki Activity

Fit PyWrap into BayesNet

Browse Source
This commit is contained in:
Ricardo Montañana Gómez
2023-11-13 11:13:32 +01:00
parent 6a23e2cc26
commit 431b3a3aa5
15 changed files with 48 additions and 40 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/PyClassifiers/PyClassifier.cc
View File

@@ -74,15 +74,15 @@ namespace pywrap {
Py_XDECREF(incoming);
return resultTensor;
}
double PyClassifier::score(torch::Tensor& X, torch::Tensor& y)
float PyClassifier::score(torch::Tensor& X, torch::Tensor& y)
{
auto [Xn, yn] = tensors2numpy(X, y);
CPyObject Xp = bp::incref(bp::object(Xn).ptr());
CPyObject yp = bp::incref(bp::object(yn).ptr());
auto result = pyWrap->score(id, Xp, yp);
float result = pyWrap->score(id, Xp, yp);
return result;
}
void PyClassifier::setHyperparameters(const nlohmann::json& hyperparameters)
void PyClassifier::setHyperparameters(nlohmann::json& hyperparameters)
{
// Check if hyperparameters are valid, default is no hyperparameters
const std::vector<std::string> validKeys = { };