diff options
Diffstat (limited to 'lj_matrix/__main__.py')
-rw-r--r-- | lj_matrix/__main__.py | 10 |
1 files changed, 9 insertions, 1 deletions
diff --git a/lj_matrix/__main__.py b/lj_matrix/__main__.py index 98f341e1e..811024ff0 100644 --- a/lj_matrix/__main__.py +++ b/lj_matrix/__main__.py @@ -24,6 +24,14 @@ from lj_matrix.do_ml import do_ml # from lj_matrix.misc import plot_benchmarks if __name__ == '__main__': - do_ml(min_training_size=1500, max_training_size=3000) + do_ml(min_training_size=1500, + max_training_size=2000, + training_increment_size=500, + test_size=None, + ljm_diag_value=None, + ljm_sigma=1.0, + ljm_epsilon=1.0, + save_benchmarks=False, + show_msgs=True) # plot_benchmarks() print('OK!') |