remove repeat
@@ -137,8 +137,6 @@ def run(gpuid, num, dim, max_iter, tol, output):
y_train = np.r_[np.zeros(num), np.ones(num)].astype(np.int32)
y_test = np.r_[np.zeros(100), np.ones(100)].astype(np.int32)
- repeat = 5
- for i in six.moves.range(repeat):
mean1 = np.random.normal(1, scale, size=dim)
mean2 = np.random.normal(-1, scale, size=dim)
means = np.stack([mean1, mean2])
@@ -161,7 +159,7 @@ def run(gpuid, num, dim, max_iter, tol, output):
y_test_pred, means, cov = \
calc_acc(X_train_gpu, y_train_gpu, X_test_gpu, y_test_gpu,
max_iter, tol, means, covariances)
- if output is not None and i == repeat - 1:
+ if output is not None:
draw(X_test_gpu, y_test_pred, means, cov, output)
print()
