Произвольная выборка данных из набора данных sklearn - PullRequest
0 голосов
/ 15 мая 2019

У меня есть объект связки из sklearn, который выглядит следующим образом.

from sklearn.datasets import load_boston
import scipy
import numpy as np

boston = load_boston()
n_samples = boston.data.shape[0]

print(boston.keys())
dict_keys(['data', 'target', 'feature_names', 'DESCR', 'filename'])

Я хочу случайным образом отобрать 30 образцов и 30 целей из данных и ключей целей.

    X, y = [np.array([boston.data[i]]), np.array([boston.target[i]) for i in np.random(choice(n_samples, 30)])
                                                            ^
SyntaxError: invalid syntax

Это все, чтобы я мог построить регрессию, используя первую функцию

slope, intercept, r_value, p_value, std_err = scipy.stats.linregress(X[:][0], y)
regression = intercept + slope*X[:][0]

boston.data и boston.target оба являются массивами numpy. Как мне это сделать?

print(type(boston.data))
<class 'numpy.ndarray'>

print(type(boston.target))
<class 'numpy.ndarray'>

1 Ответ

0 голосов
/ 15 мая 2019

У вас есть несколько опечаток (например, random.choice), и вы также перезаписываете свои массивы.Это должно работать:

x = []
y = []
for i in np.random.choice(n_samples, 30):
    x.append(boston.data[i])
    y.append(boston.target[i])
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...