Skip to content

Commit 827a328

Browse files
mariomario
authored andcommitted
reshape predictors
1 parent f09557e commit 827a328

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

code/python/cni_toolbox/HGR.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -251,7 +251,7 @@ def get_parameters(self, n_batch = 10000,
251251
results['mu_x'][batch] = cx / self.r_stimulus * max_radius * 2 - max_radius
252252
results['mu_y'][batch] = -cy / self.r_stimulus * max_radius * 2 - max_radius
253253
R = np.sqrt(results['mu_x'][batch]**2 + results['mu_y'][batch]**2)
254-
P = np.hstack((m_image, R))
254+
P = np.hstack((m_image.reshape(-1,1), R.reshape(-1,1)))
255255
results['sigma'][batch] = np.matmul(P, beta)
256256

257257
i = int(v / n_mask * 21)

0 commit comments

Comments
 (0)