This is an example code to generate classification data of two classes.
1. matplotlib inline
2. import matplotlib.pyplot as plt
3. import numpy as np
N1 = 100
N2 = 100
K = 2
sigma = 1.0
mean = [1, 1]
cov = [[sigma, 0], [0, sigma]]
x1 = np.random.multivariate_normal(mean, cov, N1)
cl = ['red'] * len(x1)
mean = [5, 5]
cov = [[sigma, 0], [0, sigma]]
x2 = np.random.multivariate_normal(mean, cov, N2)
c2 = ['blue'] * len(x2)
X = np.concatenate((x1, x2))
color = np.concatenate((cl, c2))
T = np.zeros((len(X), K))
for n in range(0, len(X)):
if n < len(x1):
T[n][0] = 1
if n >= N1 and n < len(x1) + len(x2):
T[n][1] = 1
T = T.astype(int)
plt.scatter(X[:, 0], X[:, 1], marker='o', c=color)
plt.show()