Skip to content

Commit 17657ab

Browse files
authored
Merge pull request #815 from amueller/rijn_example_cleanup
make output of rijn example a bit nicer
2 parents 29a023c + 24c4821 commit 17657ab

1 file changed

Lines changed: 1 addition & 2 deletions

File tree

examples/40_paper/2018_kdd_rijn_example.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -112,7 +112,6 @@
112112
if len(setups_evals[c].unique()) > 1 or c == performance_column]]
113113
# We are done with processing ``setups_evals``. Note that we still might have some irrelevant hyperparameters, e.g.,
114114
# ``random_state``. We have dropped some relevant hyperparameters, i.e., several categoricals. Let's check it out:
115-
print(setups_evals.dtypes)
116115

117116
# determine x values to pass to fanova library
118117
parameter_names = [pname for pname in setups_evals.columns.to_numpy() if pname != performance_column]
@@ -121,7 +120,7 @@
121120
for idx, pname in enumerate(parameter_names):
122121
try:
123122
fanova_results.append({
124-
'hyperparameter': pname if len(pname) < 35 else '[...] %s' % pname[-30:],
123+
'hyperparameter': pname.split(".")[-1],
125124
'fanova': evaluator.quantify_importance([idx])[(idx,)]['individual importance']
126125
})
127126
except RuntimeError as e:

0 commit comments

Comments
 (0)