org.deeplearning4j.text.corpora.treeparser

Class TreeVectorizer

java.lang.Object

org.deeplearning4j.text.corpora.treeparser.TreeVectorizer

public class TreeVectorizer
extends java.lang.Object

Tree vectorization pipeline. Takes a word vector model (as a lookup table) and a parser and handles vectorization of strings appropriate for an RNTN

Constructor Summary

Constructors

Constructor and Description
TreeVectorizer() Uses word vectors from the passed in word2vec model
TreeVectorizer(TreeParser parser) Uses the given parser and model for vectorization of strings

Method Summary

Modifier and Type	Method and Description
java.util.List<Tree>	getTrees(java.lang.String sentences) Vectorizes the passed in sentences
java.util.List<Tree>	getTreesWithLabels(java.lang.String sentences, java.util.List<java.lang.String> labels) Vectorizes the passed in sentences
java.util.List<Tree>	getTreesWithLabels(java.lang.String sentences, java.lang.String label, java.util.List<java.lang.String> labels) Vectorizes the passed in sentences

Methods inherited from class java.lang.Object

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

Constructor Detail

TreeVectorizer

public TreeVectorizer(TreeParser parser)

Uses the given parser and model for vectorization of strings

Parameters:

parser - the parser to use for converting strings to trees

TreeVectorizer

public TreeVectorizer()
               throws java.lang.Exception

Uses word vectors from the passed in word2vec model

Throws:

java.lang.Exception

Method Detail

getTrees

public java.util.List<Tree> getTrees(java.lang.String sentences)
                              throws java.lang.Exception

Vectorizes the passed in sentences

Parameters:

sentences - the sentences to convert to trees

Returns:

a list of trees pre converted with CNF and binarized and word vectors at the leaves of the trees

Throws:

java.lang.Exception

getTreesWithLabels

public java.util.List<Tree> getTreesWithLabels(java.lang.String sentences,
                                               java.lang.String label,
                                               java.util.List<java.lang.String> labels)
                                        throws java.lang.Exception

Vectorizes the passed in sentences

Parameters:

sentences - the sentences to convert to trees

label - the label for the sentence

labels - all of the possible labels for the trees

Returns:

a list of trees pre converted with CNF and binarized and word vectors at the leaves of the trees

Throws:

java.lang.Exception

getTreesWithLabels

public java.util.List<Tree> getTreesWithLabels(java.lang.String sentences,
                                               java.util.List<java.lang.String> labels)
                                        throws java.lang.Exception

Vectorizes the passed in sentences

Parameters:

sentences - the sentences to convert to trees

labels - all of the possible labels for the trees

Returns:

a list of trees pre converted with CNF and binarized and word vectors at the leaves of the trees

Throws:

java.lang.Exception