小编okh*_*okh的帖子

我一直在使用convnetjs 1年,现在我想继续使用更强大,更快速的库.我认为Tensorflow比JS库快几个数量级,所以我为两个库编写了一个简单的神经网络,并进行了一些测试.它是一个3-5-5-1神经网络,在一个单一的例子中训练了一定数量的具有SGD和RELU层的时期.

Tensorflow代码:

import tensorflow as tf
import numpy
import time

NUM_CORES = 1  # Choose how many cores to use.
sess = tf.Session(config=tf.ConfigProto(inter_op_parallelism_threads=NUM_CORES, intra_op_parallelism_threads=NUM_CORES))

# Parameters
learning_rate = 0.001
training_epochs = 1000
batch_size = 1

# Network Parameters
n_input = 3 # Data input
n_hidden_1 = 5 # 1st layer num features
n_hidden_2 = 5 # 2nd layer num features
n_output = 1 # Data output

# tf Graph input
x = tf.placeholder("float", [None, n_input], "a")
y = tf.placeholder("float", …