小编Esi*_*dor的帖子

简单的多层感知器模型在 TensorFlow 中不收敛

我是 TensorFlow 的新手。今天我尝试在 TF 中实现我的第一个模型,但它返回了奇怪的结果。我知道我在这里遗漏了一些东西,但我无法弄清楚。这是故事。

模型

我有一个简单的多层感知器模型,只有一个隐藏层应用于 MNIST 数据库。层定义为 [input(784) , hidden_​​layer(470) , output_layer(10)] ,tanh隐藏层的非线性和softmax输出层的损失。我使用的优化器是梯度下降算法,学习率为0.01。我的 mini batch size 是 1(我正在用样本一一训练模型)。

我的实现:

  1. 首先,我用 C++ 实现了我的模型并获得了大约 96% 的准确率。这是存储库:https : //github.com/amin2ros/Artificog
  2. 我在 TensorFlow 中实现了确切的模型,但令人惊讶的是该模型根本没有收敛。这是代码。

代码:

import sys
import input_data
import matplotlib.pyplot as plt
from pylab import *
mnist = input_data.read_data_sets("/tmp/data/", one_hot=True)
import tensorflow as tf
# Parameters
learning_rate = 0.1
training_epochs = 1
batch_size = 1
display_step = 1
# Network Parameters
n_hidden_1 = …
Run Code Online (Sandbox Code Playgroud)

deep-learning tensorflow

5
推荐指数
0
解决办法
2763
查看次数

如何在 Java 应用程序中嵌入 H2o?

我正在尝试在 Java 应用程序中启动嵌入式 H2o 并训练模型。但是我没有得到文档中的确切解释(http://docs.h2o.ai/h2o/latest-stable/h2o-docs/faq/java.html)。任何人都可以通过提供一个例子来帮助我吗?

谢谢,

java machine-learning h2o sparkling-water

1
推荐指数
1
解决办法
1787
查看次数