predict_demo.py

import tensorflow as tf
import numpy as np

from keras_train import Dataset


# 加载保存的 Keras 模型 (.h5)
model = tf.keras.models.load_model('model_checkpoints/particle_net_lite_model.001.h5')

val_dataset = Dataset('converted/val_file_0.awkd', data_format='channel_last')

predictions = model.predict(val_dataset.X)

print(predictions)