如何获取文本格式的模型预测标签?

Question

我有一个模型，我训练并保存了它，现在我只想加载它然后为我预测1000张图像文件我通过以下代码加载它

loaded_model = tf.keras.models.load_model('dir') 

我用这段代码来预测我的数据

loaded_model.predict(test_dataset)

这段代码给了我一个NumPy数组，我如何获得在model. prediction中预测的标签作为文本文件。

Answer 1

可以将字符串映射为数字，然后加入到文本

', '.join(list(array.astype(str).flatten()))
', '.join(array.astype(str).flatten().tolist()))