from sklearn.mixture import GaussianMixture
import joblib
# 训练模型并保存
gmm = GaussianMixture(n_components=5, n_init=10)
gmm.fit(DataSet)
joblib.dump(filename='gmm.model', value=gmm)
# 载入模型使用
model = joblib.load(filename='gmm.model')
y = model.predict(x)
【20211123】【机器/深度学习】使用 joblib 保存训练好的模型,并载入模型使用
于 2021-11-23 20:57:30 首次发布