diff options
author | andreas128 <Andreas> | 2017-09-26 17:47:51 +0200 |
---|---|---|
committer | andreas128 <Andreas> | 2017-09-26 17:47:51 +0200 |
commit | 80ae5c80144451706ba00b24046b31d89a7c0165 (patch) | |
tree | ab83a80569c1a0c2450349dba2e986dcc4babc26 /dpd/src/Model_Poly.py | |
parent | dd757ee779fde3a2d2a9694675de117bde4d62b4 (diff) | |
download | dabmod-80ae5c80144451706ba00b24046b31d89a7c0165.tar.gz dabmod-80ae5c80144451706ba00b24046b31d89a7c0165.tar.bz2 dabmod-80ae5c80144451706ba00b24046b31d89a7c0165.zip |
Move plot config in const.py
Diffstat (limited to 'dpd/src/Model_Poly.py')
-rw-r--r-- | dpd/src/Model_Poly.py | 5 |
1 files changed, 2 insertions, 3 deletions
diff --git a/dpd/src/Model_Poly.py b/dpd/src/Model_Poly.py index f6c024c..e799f1e 100644 --- a/dpd/src/Model_Poly.py +++ b/dpd/src/Model_Poly.py @@ -44,8 +44,7 @@ class Poly: def __init__(self, c, learning_rate_am=1.0, - learning_rate_pm=1.0, - plot=False): + learning_rate_pm=1.0): self.c = c self.learning_rate_am = learning_rate_am @@ -56,7 +55,7 @@ class Poly: self.model_am = Model_AM.Model_AM(c, plot=True) self.model_pm = Model_PM.Model_PM(c, plot=True) - self.plot = plot + self.plot = c.MDL_plot def reset_coefs(self): self.coefs_am = np.zeros(5, dtype=np.float32) |