37、数据处理与增强：构建高效机器学习数据管道

数据处理与增强：构建高效机器学习数据管道

1. 自定义层与数据管道基础

在构建数据管道时，我们可以使用自定义层对输入数据进行预处理。以下是一个使用 Layer 子类定义的自定义层 Rescaling 的示例代码：

import tensorflow as tf
from tensorflow.keras import Sequential
from tensorflow.keras.layers import Conv2D, ReLU, Activation
from tensorflow.keras.layers import BatchNormalization, Dense, Flatten
from tensorflow.keras.layers import Layer

class Rescaling(Layer):
    """ Custom Layer for Preprocessing Input """
    def __init__(self, scale, input_shape=None, name=None):
        """ Constructor """
        super(Rescaling, self).__init__(input_shape=input_shape, name=name)
        self.scale = scale

    def build(self, input_shape):
        """ Handler for building the layer """
        s