import tensorflow as tf from tensorflow.keras.models import Sequential from tensorflow.keras.layers import Dense

构建神经网络模型

model = Sequential() model.add(Dense(32, input_shape=(3,), activation='relu')) model.add(Dense(16, activation='relu')) model.add(Dense(1, activation='linear'))

编译模型

model.compile(optimizer='adam', loss='mean_squared_error')

训练模型

model.fit(X_train_reg, y_train_reg, epochs=10, batch_size=32)

预测结果

y_pred_reg = model.predict(X_test_reg)