import sys

Author: mario

code/python/cni_toolbox/HGR.py

@@ -18,6 +18,7 @@
 along with this program.  If not, see <http://www.gnu.org/licenses/>.
 '''
 
+import sys
 import numpy as np
 from scipy.stats import zscore
 from scipy.fft import fft, ifft
@@ -224,7 +225,7 @@ def get_parameters(self, n_batch = 10000,
             results['mu_x'][batch] = cx / self.r_stimulus * max_radius * 2 - max_radius
             results['mu_y'][batch] = -cy / self.r_stimulus * max_radius * 2 - max_radius
             R = np.sqrt(results['mu_x'][batch]**2 + results['mu_y'][batch]**2)
-            P = np.hstack((m_image, R))
+            P = np.hstack((m_image.reshape(-1,1), R.reshape(-1,1)))
             results['sigma'][batch] = np.matmul(P, beta)
 
             i = int(v / n_mask * 21)