org.deeplearning4j.text.invertedindex

Class DefaultInvertedIndex

java.lang.Object

org.deeplearning4j.text.invertedindex.DefaultInvertedIndex

All Implemented Interfaces:

Serializable, InvertedIndex

public class DefaultInvertedIndex
extends Object
implements InvertedIndex

Created by agibsonccc on 10/21/14.

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
DefaultInvertedIndex()

Method Summary

Modifier and Type	Method and Description
void	addWordsToDoc(int doc, List<VocabWord> words) Adds words to the given document
void	addWordToDoc(int doc, VocabWord word) Add word to a document
Collection<Integer>	allDocs() Returns a list of all documents
List<VocabWord>	document(int index) Returns a list of words for a document
List<Integer>	documents(VocabWord vocabWord) Returns the list of documents a vocab word is in
void	finish() Finishes saving data
int	numDocuments() Returns the number of documents

Methods inherited from class java.lang.Object

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

Constructor Detail

DefaultInvertedIndex

public DefaultInvertedIndex()

Method Detail

document

public List<VocabWord> document(int index)

Description copied from interface: InvertedIndex

Returns a list of words for a document

Specified by:

document in interface InvertedIndex

Returns:

documents

public List<Integer> documents(VocabWord vocabWord)

Description copied from interface: InvertedIndex

Returns the list of documents a vocab word is in

Specified by:

documents in interface InvertedIndex

Parameters:

vocabWord - the vocab word to get documents for

Returns:

the documents for a vocab word

numDocuments

public int numDocuments()

Description copied from interface: InvertedIndex

Returns the number of documents

Specified by:

numDocuments in interface InvertedIndex

Returns:

allDocs

public Collection<Integer> allDocs()

Description copied from interface: InvertedIndex

Returns a list of all documents

Specified by:

allDocs in interface InvertedIndex

Returns:

the list of all documents

addWordToDoc

public void addWordToDoc(int doc,
                         VocabWord word)

Description copied from interface: InvertedIndex

Add word to a document

Specified by:

addWordToDoc in interface InvertedIndex

Parameters:

doc - the document to add to

word - the word to add

addWordsToDoc

public void addWordsToDoc(int doc,
                          List<VocabWord> words)

Description copied from interface: InvertedIndex

Adds words to the given document

Specified by:

addWordsToDoc in interface InvertedIndex

Parameters:

doc - the document to add to

words - the words to add

finish

public void finish()

Description copied from interface: InvertedIndex

Finishes saving data

Specified by:

finish in interface InvertedIndex