Skip to content

Commit

Permalink
Missing model_dataset argument
Browse files Browse the repository at this point in the history
  • Loading branch information
stewarthe6 committed Dec 11, 2024
1 parent cd14bd9 commit 00a74af
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions atomsci/ddm/pipeline/model_wrapper.py
Original file line number Diff line number Diff line change
Expand Up @@ -905,8 +905,8 @@ def train_kfold_cv(self, pipeline):

for ei in LCTimerKFoldIterator(self.params, pipeline, self.log):
# Create PerfData structures that are only used within loop to compute metrics during initial training
train_perf_data = perf.create_perf_data(self.params.prediction_type, 'train')
test_perf_data = perf.create_perf_data(self.params.prediction_type, 'test')
train_perf_data = perf.create_perf_data(self.params.prediction_type, pipeline.data, 'train')
test_perf_data = perf.create_perf_data(self.params.prediction_type, pipeline.data, 'test')
for k in range(num_folds):
self.model = models[k]
train_dset, valid_dset = pipeline.data.train_valid_dsets[k]
Expand Down

0 comments on commit 00a74af

Please sign in to comment.