Package org.deeplearning4j.nn.adapters

Class ArgmaxAdapter

java.lang.Object

org.deeplearning4j.nn.adapters.ArgmaxAdapter

All Implemented Interfaces:

Serializable, OutputAdapter<int[]>

public class ArgmaxAdapter
extends Object
implements OutputAdapter<int[]>

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor	Description
ArgmaxAdapter()

Method Summary

Modifier and Type	Method	Description
int[]	apply(INDArray... outputs)	This method does conversion from INDArrays to int[], where each element will represents position of the highest element in output INDArray I.e.

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

ArgmaxAdapter

public ArgmaxAdapter()

Method Detail

apply

public int[] apply(INDArray... outputs)

This method does conversion from INDArrays to int[], where each element will represents position of the highest element in output INDArray I.e. Array of {0.25, 0.1, 0.5, 0.15} will return int array with length of 1, and value {2}

Specified by:

apply in interface OutputAdapter<int[]>

Parameters:

outputs -

Returns: