Make TANNew same as TAN with local discretization

This commit is contained in:
2023-08-01 13:17:12 +02:00
parent a18fbe5594
commit 8f8f9773ce
4 changed files with 39 additions and 6 deletions

View File

@@ -114,7 +114,7 @@ namespace platform {
cout << " (" << setw(5) << samples << "," << setw(3) << features.size() << ") " << flush;
// Prepare Result
auto result = Result();
auto [values, counts] = at::_unique(y);;
auto [values, counts] = at::_unique(y);
result.setSamples(X.size(1)).setFeatures(X.size(0)).setClasses(values.size(0));
int nResults = nfolds * static_cast<int>(randomSeeds.size());
auto accuracy_test = torch::zeros({ nResults }, torch::kFloat64);

View File

@@ -99,7 +99,6 @@ int main(int argc, char** argv)
filesToTest = platform::Datasets(path, true, platform::ARFF).getNames();
saveResults = true;
}
/*
* Begin Processing
*/