Fix some mistakes
This commit is contained in:
@@ -145,7 +145,6 @@ int main(int argc, char** argv)
|
||||
fold = new StratifiedKFold(n_folds, y, -1);
|
||||
else
|
||||
fold = new KFold(n_folds, y.numel(), -1);
|
||||
|
||||
auto experiment = Experiment();
|
||||
experiment.setDiscretized(discretize_dataset).setModel(model_name).setPlatform("cpp");
|
||||
experiment.setStratified(stratified).setNFolds(5).addRandomSeed(271).setScoreName("accuracy");
|
||||
|
@@ -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);
|
||||
|
Reference in New Issue
Block a user