首页 文章

如何使用由经过训练的神经网络创建的权重矩阵在另一个文件中进行预测?

提问于
浏览
-5

我已经使用keras训练了一个神经网络分类器,它目前预测的准确率为90% . 我想在不必包含整个模型的情况下对另一个文件进行预测 . 是否可以仅使用输出权重矩阵进行预测?即,在数学上,我如何使用权重矩阵进行相同的分类,而不必在每次获得需要分类预测的新数据输入时重新训练模型?

1 回答

  • 0

    保存模型或模型的权重:

    model.save(filename)
    np.save(filename, model.get_weights())
    

    对于装载,在第一种情况下:

    from keras.models import load_model
    model = load_model(filename)
    

    在第二种情况:

    #recreate the model then:
    model.set_weights(np.load(filename))
    

    然后:

    results = model.predict(batch_of_data)
    

相关问题