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
19
Source : CNN.java
with MIT License
from DRL-CASIA
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);
}