|
Java example source code file (SimpleLabelAwareIterator.java)
This example Java source code file (SimpleLabelAwareIterator.java) is included in the alvinalexander.com
"Java Source Code
Warehouse" project. The intent of this project is to help you "Learn
Java by Example" TM.
Learn more about this Java project at its project page.
The SimpleLabelAwareIterator.java Java example source code
package org.deeplearning4j.text.documentiterator;
import lombok.NonNull;
import java.util.Iterator;
/**
* This class provide option to build LabelAwareIterator from Iterable<LabelledDocument> or Iterator objects
*
* @author raver119@gmail.com
*/
public class SimpleLabelAwareIterator implements LabelAwareIterator {
protected transient Iterable<LabelledDocument> underlyingIterable;
protected transient Iterator<LabelledDocument> currentIterator;
protected LabelsSource labels = new LabelsSource();
/**
* Builds LabelAwareIterator instance using Iterable object
* @param iterable
*/
public SimpleLabelAwareIterator(@NonNull Iterable<LabelledDocument> iterable) {
this.underlyingIterable = iterable;
this.currentIterator = underlyingIterable.iterator();
}
/**
* Builds LabelAwareIterator instance using Iterator object
* PLEASE NOTE: If instance is built using Iterator object, reset() method becomes unavailable
*
* @param iterator
*/
public SimpleLabelAwareIterator(@NonNull Iterator<LabelledDocument> iterator) {
this.currentIterator = iterator;
}
/**
* This method checks, if there's more LabelledDocuments in underlying iterator
* @return
*/
@Override
public boolean hasNextDocument() {
return currentIterator.hasNext();
}
/**
* This method returns next LabelledDocument from underlying iterator
* @return
*/
@Override
public LabelledDocument nextDocument() {
LabelledDocument document = currentIterator.next();
if (document.getLabel() != null) {
labels.storeLabel(document.getLabel());
}
return document;
}
/**
* This methods resets LabelAwareIterator by creating new Iterator from Iterable internally
*/
@Override
public void reset() {
if (underlyingIterable != null)
this.currentIterator = this.underlyingIterable.iterator();
else throw new UnsupportedOperationException("You can't use reset() method for Iterator<> based instance, please provide Iterable<> instead, or avoid reset()");
}
/**
* This method returns LabelsSource instance, containing all labels derived from this iterator
* @return
*/
@Override
public LabelsSource getLabelsSource() {
return labels;
}
}
Other Java examples (source code examples)
Here is a short list of links related to this Java SimpleLabelAwareIterator.java source code file:
|