【问题标题】:Multiple regression output nodes in tensorflow learn张量流学习中的多个回归输出节点
【发布时间】:2016-10-08 17:39:37
【问题描述】:
【问题讨论】:
标签:
python
tensorflow
skflow
【解决方案1】:
似乎使用 tflearn 将是另一种选择。
更新:我意识到我们应该使用 Keras 作为 tensorflow+ theano 开发良好的 API。
【解决方案2】:
使用 tflearn 可以:
net = tfl.input_data(shape=[None, n_features1, n_features2], name='input')
net = tfl.fully_connected(net, 128, activation='relu')
net = tfl.fully_connected(net, n_features, activation='linear')
net = tfl.regression(net, batch_size=batch_size, loss='mean_square', name='target')
将此处的 128 个节点的单个全连接层替换为您想要的任何网络架构。并且不要忘记选择适合您问题的损失函数,例如用于分类的交叉熵。
python 2.7.11,张量流 0.10.0rc0,tflearn 0.2.1