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

Question

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

请告诉我如何预测。只是一个功能名称也是受欢迎的。

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)