Skip to content

Commit

Permalink
Fix issue in three sample data gen
Browse files Browse the repository at this point in the history
  • Loading branch information
tpsatish95 committed Mar 3, 2019
1 parent 6069692 commit ab4277d
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -17,7 +17,7 @@ def generate_three_two_d_gaussians(epsilon, num_samples, type=1):
if type == 2:
mean_one, mean_two, mean_three = [0, 0], [0, 0], [0, epsilon]
elif type == 3:
mean_one, mean_two, mean_three = [0, (np.sqrt(3)/3)*epsilon], [-epsilon/2, -(np.sqrt(3)/6)*epsilon], [epsilon/2, -(np.sqrt(3)/6)*epsilon], [0, (np.sqrt(3)/3)*epsilon]
mean_one, mean_two, mean_three = [0, (np.sqrt(3)/3)*epsilon], [-epsilon/2, -(np.sqrt(3)/6)*epsilon], [epsilon/2, -(np.sqrt(3)/6)*epsilon]

cov = [[1, 0], [0, 1]] # identity matrix
one = np.random.multivariate_normal(mean_one, cov, num_samples)
Expand Down

0 comments on commit ab4277d

Please sign in to comment.