Skip to content

Commit b566af5

Browse files
author
Sergey Feldman
committed
wrapup for 01
1 parent 04028c6 commit b566af5

File tree

3 files changed

+462
-10
lines changed

3 files changed

+462
-10
lines changed

01_compare_baseline_models.py

Lines changed: 0 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -22,12 +22,9 @@
2222

2323

2424
database = pd.read_json("database.json").T
25-
# note: for now we will ignore those with missing values
26-
# there are very few of them
2725
# note: this meta-dataset has swallowed information about what's categorical and what isn't
2826
# which means we are just going to be using each feature as continuous even though it
2927
# may not be
30-
database = database[database.mv == 0]
3128
database = database[database.nrow >= 50]
3229

3330

@@ -123,13 +120,6 @@ def define_and_evaluate_pipelines(X, y, random_state=0):
123120
evaluated_datasets = np.array(evaluated_datasets)
124121
times = np.array(times)
125122

126-
# remove things with exactly 1.0 score as it means it's not interesting
127-
128-
129123
# save everything to disk so we can make plots elsewhere
130124
with open("results/01_compare_baseline_models.pickle", "wb") as f:
131125
pickle.dump((results1, results2, results3, evaluated_datasets, times), f)
132-
133-
134-
# find all the datasets
135-

0 commit comments

Comments
 (0)