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[]>

This OutputAdapter implementation is suited for silent conversion of 2D SoftMax output

Author:

[email protected]

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
ArgmaxAdapter()

Method Summary

Modifier and Type	Method and 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: