因此,对于我正在制作的应用程序,我正在使用 tf.keras.models.Sequential。我知道机器学习有线性和多线性回归模型。在Sequential的文档中,据说该模型是层的线性堆栈。这等于多元线性回归吗?我能找到的关于层的线性堆栈的唯一解释是Stackoverflow 上的这个问题。
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