Как получить метки из созданного набора данных sklearn? - PullRequest
0 голосов
/ 09 июля 2019

Я создал массив с метками, используя приведенный ниже код.Когда я запускаю X[y == 0], он выдает мне только те значения в массиве X, для которых y равно 0. Однако, если у меня просто было X и я не знал, что y было именем столбца, как мне это сделать?получить имя столбца y и значения для y, которые равны 0 и 1 от X?

Что-то вроде X.column_names или X.labels - это то, что я ищу.

import torch
import numpy as np
from sklearn import datasets

X, y = datasets.make_blobs(n_samples = n_pts, random_state = 123, centers = centers, cluster_std = 0.4)
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...