chainer - 使用 Chainer 从模型中预测值的最佳实践

Question

使用Chainer，我model.pkl使用Iris 数据集( https://github.com/silwyb/train-iris ) 创建。所以我可以评估数据集，但我不知道输出预测值的最佳方式。

请告诉我如何预测。也欢迎仅使用函数名称。

Answer 1

以下作品。 silwyb/train-iris/predict.py

def predict(x_test):
    x = Variable(x_test)

    h1 = F.dropout(F.relu(model.l1(x)))
    h2 = F.dropout(F.relu(model.l2(h1)))
    y = model.l3(h2)

    return np.argmax(y.data)