org.nd4j.linalg.dataset

Class BalanceMinibatches

java.lang.Object

org.nd4j.linalg.dataset.BalanceMinibatches

public class BalanceMinibatches
extends Object

Auto balance mini batches by label.

Author:

Adam Gibson

Constructor Summary

Constructors

Constructor and Description
BalanceMinibatches()

Method Summary

Modifier and Type	Method and Description
void	balance() Generate a balanced dataset minibatch fileset.

Methods inherited from class java.lang.Object

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

Constructor Detail

BalanceMinibatches

public BalanceMinibatches()

Method Detail

balance

public void balance()

Generate a balanced dataset minibatch fileset.