Я хочу выделить два набора данных Circle - PullRequest
0 голосов
/ 03 ноября 2019

https://medium.com/@sachinkun21/using-a-linear-model-to-deal-with-nonlinear-dataset-c6ed0f7f3f51

Я хочу создать 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_circles(n_samples=n_samples, 
factor=.5, noise=.05)

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()
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...