# -*- coding: utf-8 -*-
"""
Created on Tue Aug 29 14:53:02 2017
分布式Estimator自定义模型求iris分类
@author: miclover
"""
import tensorflow as tf
from tensorflow.contrib import layers
from tensorflow.contrib import learn
from sklearn import datasets,cross_validation
def my_model(features,target):
target = tf.one_hot(target,3,1,0)
features = layers.stack(features,layers.fully_connected,[10,20,10])
prediction,loss = tf.contrib.learn.models.logistic_regression_zero_init(features,target)
train_op = tf.contrib.layers.optimize_loss(
loss,tf.contrib.framework.get_global_step(),optimizer='Adagrad',
learning_rate = 0.1)
return{'class':tf.arg_max(prediction,1),'prob':prediction},loss,train_op
iris = datasets.load_iris()
x_train,x_test,y_train,y_test = cross_validation.train_test_split(
iris.data,iris.target,test_size = 0.2,random_state = 35)
classifier = learn.Estimator(model_fn = my_model)
classifier.fit(x_train,y_train,steps = 700)
predictions = classifier.predict(x_test)
print(predictions)
没有合适的资源?快使用搜索试试~ 我知道了~
资源推荐
资源详情
资源评论
收起资源包目录
分布式Estimator自定义模型求iris分类.rar (1个子文件)
分布式Estimator自定义模型求iris分类
main.py 1KB
共 1 条
- 1
资源评论
Easy.miclover
- 粉丝: 74
- 资源: 7
上传资源 快速赚钱
- 我的内容管理 展开
- 我的资源 快来上传第一个资源
- 我的收益 登录查看自己的收益
- 我的积分 登录查看自己的积分
- 我的C币 登录后查看C币余额
- 我的收藏
- 我的下载
- 下载帮助
安全验证
文档复制为VIP权益,开通VIP直接复制
信息提交成功