代码之家  ›  专栏  ›  技术社区  ›  MysteryGuy

如何使泛型函数适应模型?

  •  0
  • MysteryGuy  · 技术社区  · 6 年前

    我正在开发一个有线电视新闻网与凯拉斯

    preprocess_input Keras模型的功能(VGG16、VGG19等) here

    所以我在做类似的事情

    preprocess_input (inputs)
    

    但是,我想测试几个模型,结果不同 预处理输入 在我的职责范围内。我怎样才能在自制的函数中说我想使用 预处理输入

    我想把我的模型名作为我函数的一个参数,但是我不知道如何将这个字符串转换成允许调用的方式

    modelThatIwant.preprocess_input(...)
    

    1 回复  |  直到 6 年前
        1
  •  1
  •   MysteryGuy    6 年前

    实际上,答案很简单:你只需要添加一个参数 fct 在您创建的自制函数中,然后传递 preprocessing_inputs 对应你想训练的模型。。。感谢@Matias Valdenegro