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

2024-09-28 22:21:03 发布

您现在位置:Python中文网/ 问答频道 /正文

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

我定义了一个函数,在这里我调用Keras模型的preprocess_input函数(VGG16,VGG19,…)here

所以我在做类似的事情

preprocess_input (inputs)

但是,我想测试几个模型,从而在我的函数中产生不同的preprocess_input。如何在我的自制函数中说我想使用与我正在测试的模型对应的preprocess_input

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

modelThatIwant.preprocess_input(...)

谢谢你的帮助


Tags: 函数字符串模型input参数here定义方式
1条回答
网友
1楼 · 发布于 2024-09-28 22:21:03

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

相关问题 更多 >