summaryrefslogtreecommitdiffstats
path: root/dpd/main.py
diff options
context:
space:
mode:
Diffstat (limited to 'dpd/main.py')
-rwxr-xr-xdpd/main.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/dpd/main.py b/dpd/main.py
index fcf1989..b4355c3 100755
--- a/dpd/main.py
+++ b/dpd/main.py
@@ -173,8 +173,8 @@ while i < num_iter:
# Model
elif state == "model":
- dpddata = model_poly.get_dpd_data(tx, rx, phase_diff)
- del extStat
+ dpddata = model.train(tx, rx, phase_diff)
+ dpddata = model.get_dpd_data()
extStat = ExtractStatistic.ExtractStatistic(c, plot=True)
state = "adapt"