使用tensorflow对Mnist数据集进行字体识别

来源:互联网 发布:点击链接直接下载 js 编辑:程序博客网 时间:2024/05/16 08:26



上代码:

#!/usr/bin/env python#coding:utf-8import tensorflowimport pandas as pdimport skflowtrain = pd.read_csv('~/Mnist/train.csv')X_train = train.drop('label',1)y_train = train['label']classifier = skflow.TensorFlowLinearClassifier(n_classes=10,batch_size=100,steps=1000,learning_rate=0.01)classifier.fit(X_train,y_train)linear_y_predict = classifier.predict(X_test)
#save results for submission onto Kagglelinear_submission = pd.DataFrame({'ImageId':range(1,28001),'label':linear_y_predict})linear_submission.to_csv('~/Mnist/linear_submission.csv',index=False)


0 0
原创粉丝点击