我有一个1000万的数据集,每个记录有4个参数。
如果我在执行model.predict
时使用以下模型进行建模,我会得到相同的结果。
# create model
model = Sequential()
model.add(Dense(12,input_dim=4, kernel_initializer='uniform', activation='tanh'))
model.add(Dense(8, kernel_initializer='uniform', activation='tanh'))
model.add(Dense(1, kernel_initializer='uniform', activation='sigmoid'))
# Compile model
model.compile(loss='binary_crossentropy', optimizer='adam', metrics=['accuracy'])
# Fit the model
model.fit(numpy.array(X1), numpy.array(Y1), epochs=50, batch_size=1000000, verbose=2)