skflow也即是 tf.contrib.learn, 是 TensorFlow 官方提供的另外一个对 TensorFlow 的高层封装,通过这个封装,用户可以和使用 sklearn 类似的方法使用 TensorFlow。
参考:Introduction to Scikit Flow
另一个高层封装 TFLearn 进一步简化了 tf.contrib.learn 中对模型定义的方法,并提供了一些更加简洁的方法来定义神经网络的结构。更多关于 TFLearn 的用法介绍可以参考 TFLearn 的官方网站(http://tflearn.org/)
一、Linear
1、skflow.TensorFlowLinearRegressor
结果:
2、skflow.TensorFlowLinearClassifier
结果:
二、DNN
1、skflow.TensorFlowDNNClassifier
结果:
2、skflow.TensorFlowDNNRegressor
结果: