首页 > 其他 > 详细

mlp_clf_mnist_test

时间:2019-02-16 22:22:16      阅读:341      评论:0      收藏:0      [点我收藏+]

import os
os.environ[‘CUDA_VISIBLE_DEVICES‘] = "0"
from mlp_clf import MLPClassifier
import numpy as np
import tensorflow as tf
from sklearn.datasets import load_svmlight_file
from sklearn.utils import shuffle
#from scipy.sparse import csr_matrix
def mean(numbers): #计算平均值
s = 0.0
for num in numbers:
s = s + num
return s/len(numbers)
def dev(numbers, mean): #计算方差
sdev = 0.0
for num in numbers:
sdev = sdev + (num - mean)**2
return pow(sdev / (len(numbers)-1), 0.5)


if __name__ == ‘__main__‘:
mze_array=[]
mae_array=[]
mse_array=[]

trainfile = "test_housing.0"
testfile = "train_housing.0"

X_train0, y_train = load_svmlight_file(trainfile)
X_test0 , y_test = load_svmlight_file(testfile)
X_train = X_train0.todense()
X_test = X_test0.todense()
attribute = np.shape(X_train)[-1]#tf.shape(X_train)
n_class = tf.reduce_max(y_train)
print("n_class:", n_class)


X_train, y_train = shuffle(X_train, y_train, random_state=0)

y_train = y_train - 1
y_test = y_test - 1

 

clf = MLPClassifier(attribute, 5, [100]*3)
log = clf.fit(X_train, y_train, n_epoch=1000, keep_prob=0.8, val_data=(X_test, y_test))
Y_pred = clf.predict(X_test)
mze = 1- (Y_pred == y_test).mean()
mae = np.abs(Y_pred.ravel()-y_test.ravel()).astype(float).mean()
mse = np.power(Y_pred.ravel()-y_test.ravel(),2).astype(float).mean()

print("mze: %.4f" % mze)
print("mae:", mae)
print("mse:", mse)

 

mlp_clf_mnist_test

原文:https://www.cnblogs.com/huadongw/p/10389332.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!