Fix some mistakes

This commit is contained in:
2023-07-26 12:53:01 +02:00
parent 099b4bea09
commit 4a54bd42a2
5 changed files with 12 additions and 7 deletions

View File

@@ -48,7 +48,6 @@ void StratifiedKFold::build()
{
stratified_indices = vector<vector<int>>(k);
int fold_size = n / k;
int remainder = n % k;
// Compute class counts and indices
auto class_indices = map<int, vector<int>>();
vector<int> class_counts(*max_element(y.begin(), y.end()) + 1, 0);