cn.centipede.numpy.Numpy.np.argmax()

Here are the examples of the java api cn.centipede.numpy.Numpy.np.argmax() taken from open source projects. By voting up you can indicate which examples are most useful and appropriate.

1 Examples 7

19 Source : CNN.java
with MIT License
from DRL-CASIA

public int predict(NDArray X) {
    X = X.reshape(np.newaxis, np.ALL);
    NDArray predict = conv1.forward(X);
    predict = relu1.forward(predict);
    predict = pool1.forward(predict);
    predict = conv2.forward(predict);
    predict = relu2.forward(predict);
    predict = pool2.forward(predict);
    predict = predict.reshape(1, -1);
    predict = nn.forward(predict);
    predict = softmax.predict(predict);
    return np.argmax(predict);
}