Add torch methods to discretize

Add fit_transform methods
This commit is contained in:
2024-06-07 23:54:42 +02:00
parent 633aa52849
commit e205668906
14 changed files with 183 additions and 87 deletions

View File

@@ -4,8 +4,8 @@
using namespace std;
namespace mdlp {
Metrics::Metrics(labels_t& y_, indices_t& indices_): y(y_), indices(indices_),
numClasses(computeNumClasses(0, indices.size()))
Metrics::Metrics(labels_t& y_, indices_t& indices_) : y(y_), indices(indices_),
numClasses(computeNumClasses(0, indices_.size()))
{
}