时间戳

首页 > 解决方案 > 如何将字符串转换为张量流模型？

python - 如何将字符串转换为张量流模型？

问题描述

所以，我已经创建了这个模型model1，我正在尝试创建一个函数，它接受不同的模型，比如model1 model2etc 并评估它们。在此过程中，我有一个名为的字符串"model1"，我想将其转换为，model1以便可以将.evaluate()方法传递给它。这可能吗？

标签： pythontensorflow

解决方案

假设您之前训练的模型已保存到磁盘，您可能只想将它们加载回内存并评估为：

model_names = ["model1", "model2", "model3", ..., "model10"]
for model_name in model_names:
    model = tf.keras.models.load_model("path/to/location/" + model_name)
    score = model.evaluate(X_test, y_test, ...)

此外，如果您计划拥有大量具有顺序名称的模型，例如 model1、model2、...、model_n，您可以使用列表推导而不是对模型名称进行硬编码，例如：

model_names = ["model" + str(x) for x in range(1, n)]

请注意，这些是伪代码，您需要根据您的需要完成...。

推荐阅读