Как использовать fit_transform с массивом?

#python-3.x #scikit-learn

Вопрос:

Пример содержимого массива:

 [
    [4.9, 3.0, 1.4, 0.2, 0.0, 2.0], 
    [4.7, 3.2, 1.3, 0.2, 0.0, 2.0], 
    [4.6, 3.1, 1.5, 0.2, 0.0, 2.0], 
    ... 
]
 
 model = TSNE(learning_rate=100)
transformed = model.fit_transform(data)
 

Я пытаюсь применить tSNE к массиву с плавающей точкой, но получаю ошибку. Что я должен изменить?

ValueError: setting an array element with a sequence. The requested array has an inhomogeneous shape after 1 dimensions. The detected shape was (149,) inhomogeneous part.

Ответ №1:

Попробуйте этот пример:

 from sklearn.manifold import TSNE
import numpy as np
X = np.array([[4.9, 3.0, 1.4, 0.2, 0.0, 2.0], [4.7, 3.2, 1.3, 0.2, 0.0, 2.0]])
model = TSNE(learning_rate=100)
transformed = model.fit_transform(X)
print(transformed)