基础 HyperModel 类
HyperModel 类
keras_tuner.HyperModel(name=None, tunable=True)
定义模型的搜索空间。
搜索空间是模型的集合。build 函数将使用给定的 HyperParameters 对象从空间中构建其中一个模型。
用户应该通过覆盖 build() 来子类化 HyperModel 类以定义其搜索空间，build() 创建并返回 Keras 模型。或者，您也可以覆盖 fit() 以自定义模型的训练过程。
示例
在 build() 中，您可以使用超参数创建模型。
class MyHyperModel(kt.HyperModel):
def build(self, hp):
model = keras.Sequential()
model.add(keras.layers.Dense(
hp.Choice('units', [8, 16, 32]),
activation='relu'))
model.add(keras.layers.Dense(1, activation='relu'))
model.compile(loss='mse')
return model
覆盖 HyperModel.fit() 时，如果使用 model.fit() 训练模型，它会返回训练历史记录，您可以直接返回它。您可以使用 hp 指定任何要调整的超参数。
class MyHyperModel(kt.HyperModel):
def build(self, hp):
...
def fit(self, hp, model, *args, **kwargs):
return model.fit(
*args,
epochs=hp.Int("epochs", 5, 20),
**kwargs)
如果您有一个自定义的训练过程，您可以将目标值作为浮点数返回。
如果要跟踪更多指标，可以返回一个要跟踪的指标字典。
class MyHyperModel(kt.HyperModel):
def build(self, hp):
...
def fit(self, hp, model, *args, **kwargs):
...
return {
"loss": loss,
"val_loss": val_loss,
"val_accuracy": val_accuracy
}
参数
- name：可选字符串，此 HyperModel 的名称。
- tunable：布尔值，指示此超模型中定义的超参数是否应添加到搜索空间中。如果为
False，则必须预先定义这些参数的搜索空间，否则将使用默认值。默认为 True。
build 方法
HyperModel.build(hp)
构建模型。
参数
- hp：一个
HyperParameters实例。
返回值
模型实例。