help messages and white space
@@ -21,10 +21,12 @@ def make_design(n_cond, n_part):
Args:
n_cond (int): Number of conditions
n_part (int): Number of partitions
+
Returns:
Tuple (cond_vec, part_vec)
cond_vec (np.ndarray): n_obs vector with condition
part_vec (np.ndarray): n_obs vector with partition
+
"""
p = np.array(range(0, n_part))
c = np.array(range(0, n_cond))
@@ -56,6 +58,7 @@ def make_dataset(model, theta, cond_vec, n_channel=30, n_sim=1,
covariance is specified
Returns:
data (list): List of rsa.Dataset with obs_descriptors
+
"""
# Get the model prediction and build second moment matrix
@@ -108,11 +111,14 @@ def make_dataset(model, theta, cond_vec, n_channel=30, n_sim=1,
def make_exact_signal(G, n_channel):
"""
Generates signal exactly with a specified second-moment matrix (G)
+
Args:
G(np.array): desired second moment matrix (ncond x ncond)
n_channel (int) : Number of channels
+
Returns:
np.array (n_cond x n_channel): random signal
+
"""
# Generate the true patterns with exactly correct second moment matrix
n_cond = G.shape[0]
