update for changes to dit interface

legao0429 · Feb 25, 2019 · 24d14e0 · 24d14e0
1 parent eea10f8
commit 24d14e0
Show file tree

Hide file tree

Showing 4 changed files with 12 additions and 11 deletions.
diff --git a/mme2.py b/mme2.py
@@ -5,7 +5,7 @@
 import dit
 import scipy.io
 import itertools
-from dit.algorithms.scipy_optimizers import maxent_dist
+from dit.algorithms.distribution_optimizers import maxent_dist
 
 fname = sys.argv[1]
 dat = sp.io.loadmat(fname)

diff --git a/mme3pred.py b/mme3pred.py
@@ -4,7 +4,8 @@
 import scipy as sp
 import dit
 import scipy.io
-from dit.algorithms.scipy_optimizers import maxent_dist
+from dit.algorithms.distribution_optimizers import maxent_dist
+
 
 fname = sys.argv[1]
 dat = sp.io.loadmat(fname)

diff --git a/pidbroja.py b/pidbroja.py
@@ -4,7 +4,7 @@
 import scipy as sp
 import dit
 import scipy.io
-from dit.algorithms.scipy_optimizers import pid_broja
+from dit.pid import PID_BROJA
 
 
 fname = sys.argv[1]
@@ -17,13 +17,13 @@
 
 d = dit.Distribution(*zip(*np.ndenumerate(P)))
 
-x = pid_broja(d, [[0],[1]], [2])
+x = PID_BROJA(d, [[0],[1]], [2])
 
 pid = np.zeros(4)
-pid[0] = x.R
-pid[1] = x.U0
-pid[2] = x.U1
-pid[3] = x.S
+pid[0] = x.get_partial(((0,), (1,)))
+pid[1] = x.get_partial(((0,),))
+pid[2] = x.get_partial(((1,),))
+pid[3] = x.get_partial(((0,1),))
 
 dat['pid'] = pid
 sp.io.savemat(fname,dat)
diff --git a/pidbrojadist.py b/pidbrojadist.py
@@ -4,11 +4,11 @@
 import scipy as sp
 import dit
 import scipy.io
-from dit.algorithms.scipy_optimizers import BROJAOptimizer
+from dit.algorithms.distribution_optimizers import BROJABivariateOptimizer
 
 def pid_broja_dist(dist, sources, target, rv_mode=None):
-    broja = BROJAOptimizer(dist, sources, target, rv_mode)
-    broja.optimize()
+    broja = BROJABivariateOptimizer(dist, sources, target, rv_mode)
+    broja.optimize(niter=10)
     opt_dist = broja.construct_dist()
     return opt_dist