python训练一个简单的回归模型保存并调用

import tensorflow as tf
import keras
import numpy as np
from keras import layers
import matplotlib.pyplot as plt
import joblib

x = np.linspace(0, 100, 30)#主要用来创建等差数列,等差间隔为30
y = 3*x+7+np.random.random(30)

print(x)
print(y)

plt.scatter(x, y)

model = keras.Sequential()
model.add(layers.Dense(1, input_dim=1)) #加入dense层，输入输出都为1维
model.summary()

#编译模型
model.compile(optimizer='adam', loss='binary_crossentropy')#binary_crossentropy为交叉熵损失函数

#训练模型
model.fit(x, y, epochs=300)
predict_y = model.predict(x)


joblib.dump(model,'D:/python/workspace/clf.pkl') #将clf存入.pkl的文件中

#提取模型
clf = joblib.load('D:/python/workspace/clf.pkl')