Skip to content

Commit 8452873

Browse files
Update regression-example-ames-no-preproc.py
Test with cast to dataset with a tensor.
1 parent b033031 commit 8452873

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

regression-example-ames-no-preproc.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -114,7 +114,8 @@
114114
# use_multiprocessing_for_multiple_neural_networks=False, # pull this param
115115
model_graphs='model_graphs',
116116
batch_size=batch_size,
117-
meta_trial_number=meta_trial_number)
117+
meta_trial_number=meta_trial_number,
118+
cast_data_to_dataset=True)
118119
result = cerebros.run_random_search()
119120

120121
print("Best model: (May need to re-initialize weights, and retrain with early stopping callback)")

0 commit comments

Comments
 (0)