【发布时间】:2019-09-24 01:33:47
【问题描述】:
所以对于我正在制作的应用程序,我正在使用 tf.keras.models.Sequential。我知道机器学习有线性和多线性回归模型。在 Sequential 的the documentation 中说模型是层的线性堆叠。这等于多元线性回归吗?我能找到的关于层的线性堆栈的唯一解释是 Stackoverflow 上的 this 问题。
def trainModel(bow,unitlabels,units):
x_train = np.array(bow)
print("X_train: ", x_train)
y_train = np.array(unitlabels)
print("Y_train: ", y_train)
model = tf.keras.models.Sequential([
tf.keras.layers.Dense(256, activation=tf.nn.relu),
tf.keras.layers.Dropout(0.2),
tf.keras.layers.Dense(len(units), activation=tf.nn.softmax)])
model.compile(optimizer='adam',
loss='sparse_categorical_crossentropy',
metrics=['accuracy'])
model.fit(x_train, y_train, epochs=50)
return model
【问题讨论】:
标签: machine-learning keras tf.keras machine-learning-model