在函数中使用keras神经网络

import keras
from keras.models import Sequential
from keras.layers import Dense

def build_model():
    model = Sequential()
    # 添加输入层和隐藏层
    model.add(Dense(units=64, activation='relu', input_dim=input_dim))
    # 添加输出层
    model.add(Dense(units=output_dim, activation='softmax'))
    return model

model.compile(loss='categorical_crossentropy', optimizer='adam', metrics=['accuracy'])

model.fit(x_train, y_train, epochs=10, batch_size=32)

predictions = model.predict(x_test)