一、代码中的数据集下载地址如下:
百度网盘提取码:lala
二、代码运行环境:
Tensorflow-gpu==2.4.0
Python==3.7
三、训练代码如下:
import tensorflow as tf
import pandas as pd
import os
os.environ['TF_XLA_FLAGS'] = '--tf_xla_enable_xla_devices'
os.environ['TF_FORCE_GPU_ALLOW_GROWTH'] = 'true'
data = pd.read_csv(r'dataset/logic_dataset.csv')
x = data.iloc[:, :-1]
y = data.iloc[:, -1].replace(-1, 0)
model = tf.keras.Sequential()
model.add(tf.keras.layers.Dense(4, input_shape=(15,), activation='relu'))
model.add(tf.keras.layers.Dense(4, activation='relu'))
model.add(tf.keras.layers.Dense(1, activation='sigmoid'))
model.summary()
model.compile(
optimizer='adam',
loss='binary_crossentropy',
metrics=['acc']
)
history = model.fit(x, y, epochs=10000)
model.save(r'model_data/model.h5')