我得到一个断言错误:
shapes do not match along axis 0: (107, 13); (0, 535)
同时运行以下代码进行分类。你知道吗
n_samples=535, n_features=13, n_classes=7
任何帮助都将不胜感激。你知道吗
import theanets
from sklearn.metrics import confusion_matrix
import scipy.io
import numpy
X = scipy.io.loadmat('berlinFeaturesCAFE.mat')
X_F = X['featureContainer'];
X_F_A = numpy.require(X_F, dtype=numpy.float32, requirements=None)
y = scipy.io.loadmat('convertLabel.mat')
y_F = y['xdNew']
cut = int(len(X_F_A) * 0.8) # training / validation split
train = X_F_A[:cut], y_F[:cut]
valid = X_F_A[cut:], y_F[cut:]
net = theanets.Classifier([13, 7])
# Train the model using SGD with momentum.
net.train(train, valid, algo='sgd', learning_rate=1e-4, momentum=0.9)
# Show confusion matrices on the training/validation splits.
for label, (X, y) in (('training:', train), ('validation:', valid)):
print(label)
print(confusion_matrix(y, net.predict(X)))
目前没有回答
相关问题 更多 >
编程相关推荐