如何在Swift中为使用coremltools转换的LSTM Keras模型形成输入

2024-05-19 03:21:43 发布

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

我使用keras LSTM模型进行分类,然后借助苹果的CoreML工具将其导出。我将模型的输入形状指定为(173,3):

model = Sequential()
model.add(LSTM(256, input_shape=(173, 3)))
model.add(Dense(1, activation='sigmoid'))

培训后,我通过coremltools保存它

coreml_model = coremltools.converters.keras.convert(model)

并获得以下模型说明:

Model description

在XCode中,它如下所示:

xcode model description

在swift中,我有3个数组,每个数组有173个值,但我不知道如何将其提供给我拥有的模型。也许我必须以不同的方式保存模型


Tags: 工具模型苹果addinputmodel分类数组

热门问题