diff options
Diffstat (limited to 'dpd/src/Model_PM.py')
| -rw-r--r-- | dpd/src/Model_PM.py | 7 |
1 files changed, 6 insertions, 1 deletions
diff --git a/dpd/src/Model_PM.py b/dpd/src/Model_PM.py index 6639382..8d35051 100644 --- a/dpd/src/Model_PM.py +++ b/dpd/src/Model_PM.py @@ -66,7 +66,7 @@ class Model_PM: ax.scatter(tx_dpd, phase_diff, label="Binned Data", color="blue", - s=0.1) + s=1) ax.set_title("Model_PM") ax.set_xlabel("TX Amplitude") ax.set_ylabel("Phase DIff") @@ -76,6 +76,10 @@ class Model_PM: fig.savefig(fig_path) plt.close(fig) + def _discard_small_values(self, tx_dpd, phase_diff): + mask = tx_dpd > self.c.MPM_tx_min + return tx_dpd[mask], phase_diff[mask] + def poly(self, sig): return np.array([sig ** i for i in range(0, 5)]).T @@ -88,6 +92,7 @@ class Model_PM: return tx_range, phase_diff def get_next_coefs(self, tx_dpd, phase_diff, coefs_pm): + tx_dpd, phase_diff = self._discard_small_values(tx_dpd, phase_diff) check_input_get_next_coefs(tx_dpd, phase_diff) coefs_pm_new = self.fit_poly(tx_dpd, phase_diff) |