卷积神经网络输出值在范围内

Question

我正在尝试使用 tensorflow 学习卷积神经网络，我正在使用下面的代码来创建我的网络。

    # network weights
    W_conv1 = weight_variable([8, 8, 4, 32])
    b_conv1 = bias_variable([32])

    W_conv2 = weight_variable([4, 4, 32, 64])
    b_conv2 = bias_variable([64])

    W_conv3 = weight_variable([3, 3, 64, 64])
    b_conv3 = bias_variable([64])

    W_fc1 = weight_variable([1600, 512])
    b_fc1 = bias_variable([512])

    W_fc2 = weight_variable([512, ACTIONS])
    b_fc2 = bias_variable([ACTIONS])

    # input layer
    s = tf.placeholder("float", [None, 80, 80, 4])

    # hidden layers
    h_conv1 = tf.nn.relu(conv2d(s, W_conv1, 4) + b_conv1)
    h_pool1 = max_pool_2x2(h_conv1)

    h_conv2 = tf.nn.relu(conv2d(h_pool1, W_conv2, 2) + b_conv2)
    #h_pool2 = max_pool_2x2(h_conv2)

    h_conv3 = tf.nn.relu(conv2d(h_conv2, W_conv3, 1) + b_conv3)
    #h_pool3 = max_pool_2x2(h_conv3)

    #h_pool3_flat = tf.reshape(h_pool3, [-1, 256])
    h_conv3_flat = tf.reshape(h_conv3, [-1, 1600])

    h_fc1 = tf.nn.relu(tf.matmul(h_conv3_flat, W_fc1) + b_fc1)

    # readout layer
    readout = tf.matmul(h_fc1, W_fc2) + b_fc2

此网络采用 [80,80,4] 对象输出 ACTIONS 预测；（ACTIONS 类数）

我的问题是如何创建一个输出 [a,b] 之间值的网络

Answer 1

如果 [a,b] 是分类的，即你有离散的类（比如总共 n 个），你需要在读出层有 n 个神经元，每个对应于一个给定的类，代表它的概率。标准方法是在 logits 上使用 sigmoid 交叉熵（在 tensorflow 的情况下使用 tf.nn.sigmoid_cross_entropy_with_logits）。

如果 [a,b] 是连续的（我认为这可能不是您想要做的），那么它就会成为回归问题。因此，给定足够的数据和适当的损失函数（平方误差）以及一些正则化应该确保您的输出在可接受的范围内，或者您始终可以将一个数字范围映射到另一个范围。

如果这些都没有意义，那么在深入研究神经网络之前，您需要学习一些基本的机器学习。