我希望为回归问题部署一个50模型集合,每个模型都是连续煤焦神经网络。在
下面是我代码的一个(简化为3个模型)版本,运行良好,工作正常。在
但是,我不想为每个单独的模型创建pickle文件,那么有没有一种方法可以创建一个包含所有模型列表的类,从而只需要保存/加载一个pickle文件?在from __future__ import absolute_import, division, print_function
import tensorflow as tf
from tensorflow import keras
import pandas as pd
import numpy as np
train = pd.read_csv("Training Data.csv").fillna(0)
X_train = train.drop(['ID_NUMBER','DATE','X','Y','Z'],axis=1)
Y_train = train[['X','Y','Z']]
EPOCHS = 1500
BATCH_SIZE = 256
#Defining the 3 layered Neural Network
def build_model():
model = keras.Sequential([
keras.layers.Dense(1000, activation=tf.nn.softplus,
input_shape=(X_train.shape[1],)),
keras.laye