27、机器学习最佳实践与卷积神经网络在服装图像分类中的应用

机器学习最佳实践与卷积神经网络在服装图像分类中的应用

1. TensorFlow 模型的保存与恢复

在 TensorFlow 中，我们可以按照以下步骤保存并恢复模型。这里以在癌症数据集上训练一个简单的逻辑回归模型为例：
1. 导入必要的 TensorFlow 模块并加载癌症数据集 ：

import tensorflow as tf
from tensorflow import keras
from sklearn import datasets
cancer_data = datasets.load_breast_cancer()
X = cancer_data.data
Y = cancer_data.target

2. 使用 Keras 顺序 API 构建简单的逻辑回归模型 ：

learning_rate = 0.005
n_iter = 10
tf.random.set_seed(42)
model = keras.Sequential([
    keras.layers.Dense(units=1, activation='sigmoid')
])
model.compile(loss='binary_crossentropy',
              optimizer=tf.keras.optimizers.Adam(learning_rate))