Я хочу разделить набор данных двух лун. Я хочу создать NN, который классифицирует две группы и оценивает производительность, используя тестовые данные. Что я могу сделать, чтобы завершить код? Пожалуйста, дайте мне полный код.
я использовал 'colab'
import numpy as np
import matplotlib.pyplot as plt
from sklearn import cluster, datasets
np.random.seed(0)
n_samples = 1500
(all_data, all_labels) = datasets.make_moons(n_samples=n_samples,
noise=.1)
data_train = all_data[ :10000] # 1000 data for training
data_test = all_data[ 10000:] # 500 data for test
label_train = all_labels[ :10000] # 1000 labels for training
label_test = all_labels[ 10000:] # 500 labels for test
plt.figure(figsize=(12,8))
mask=label_train>0
plt.scatter(data_train[~mask,0],data_train[~mask,1], color='b', label =
'class 0')
plt.scatter(data_train[mask,0],data_train[mask,1], color='r', label =
'class 1')
plt.axis('equal')
plt.show()