Keras ASR 模型验证步骤

Question

我是机器学习的新手。我想设计一个ASR（自动语音识别）的模型，所以我遇到了以下链接：

https://keras.io/examples/audio/transformer_asr/

我能够理解训练模型的概念。我还使用以下代码保存了上述模型：

model.save_weights('data_sa',save_format='hdf5')

然后我再次使用：

model.load_weights('/content/data_sa')

现在，我想通过给出验证数据集来预测模型。但我遇到了一个问题。

我正在使用以下步骤进行预测：

model.predict(np.array(val_ds)) #this is the validation dataset

它给出了张量转换的错误。

我尝试搜索此步骤的可能解决方案，但找不到。

Answer 1

tf.cast() 和 tf.to_float() 是 TensorFlow 函数，因此您可以使用 import tensorflow as tf 来获取它们。