【发布时间】:2021-04-14 04:08:53
【问题描述】:
使用以下网站: https://scikit-learn.org/stable/auto_examples/manifold/plot_lle_digits.html#sphx-glr-auto-examples-manifold-plot-lle-digits-py https://scikit-learn.org/stable/auto_examples/manifold/plot_swissroll.html#sphx-glr-auto-examples-manifold-plot-swissroll-py
我设法在 MNIST 数据集和 swissroll 数据集上获得了 LLE,但不知何故,我不明白如何才能让它在像 https://www.kaggle.com/manufacturingai/predicting-fraud-w-fast-ai 这样的外部数据集上运行。
我的尝试如下:
import numpy as np
import matplotlib.pyplot as plt
import pandas as pd
%matplotlib inline
from matplotlib import offsetbox
from sklearn import (manifold, datasets)
n_neighbors = 30
f_fontsize = 8
data = np.genfromtxt('../content/creditcard.csv', skip_header=True)
features = data[:, :3]
targets = data[:, 3] # The last column is identified as the target
def plotcreditfraudfig(X, color, X_sr, err):
fig = plt.figure()
ax = fig.add_subplot(211, projection='3d')
ax.scatter(X[:, 0], X[:, 1], X[:, 2],cmap=plt.cm.Spectral)
ax.set_title("Original data")
ax = fig.add_subplot(212)
ax.scatter(X_sr[:, 0], X_sr[:, 1],cmap=plt.cm.Spectral)
plt.axis('tight')
plt.xticks([]), plt.yticks([])
plt.title('Projected data')
plt.show()
clf = manifold.LocallyLinearEmbedding(n_neighbors=n_neighbors, n_components=2, method='standard')
clf.fit(X=features, y=targets)
print("Done. Reconstruction error: %g" %clf.reconstruction_error_)
X_llecf=clf.transform(X)
plot_embedding(X_llecf, "Locally Linear Embedding")
---------------------------------------------------------------------------
IndexError Traceback (most recent call last)
<ipython-input-106-91224a1ba194> in <module>()
1 data = np.genfromtxt('../content/creditcard.csv', skip_header=True)
----> 2 features = data[:, :3]
3 targets = data[:, 3] # The last column is identified as the target
4
5 clf = manifold.LocallyLinearEmbedding(n_neighbors=n_neighbors, n_components=2, method='standard')
IndexError: too many indices for array: array is 1-dimensional, but 2 were indexed
【问题讨论】:
-
您似乎在创建“功能”时遇到了问题。尝试绘制数据 head() 并查看您的数据如何。您的问题似乎在您的数据集中。你需要一个数据框来做你在做什么
标签: python scikit-learn dimensionality-reduction