public WordEmbeddingsEstimator(IHostEnvironment env,
WordEmbeddingsTransform.PretrainedModelKind modelKind = WordEmbeddingsTransform.PretrainedModelKind.Sswe, params WordEmbeddingsTransform.ColumnInfo[] columns)
{
Contracts.CheckValue(env, nameof(env));
_host = env.Register(nameof(WordEmbeddingsEstimator));
_modelKind = modelKind;
_customLookupTable = null;
_columns = columns;
}
public Reconciler(WordEmbeddingsTransform.PretrainedModelKind modelKind = WordEmbeddingsTransform.PretrainedModelKind.Sswe)
{
_modelKind = modelKind;
_customLookupTable = null;
}
public OutColumn(VarVector <string> input, WordEmbeddingsTransform.PretrainedModelKind modelKind = WordEmbeddingsTransform.PretrainedModelKind.Sswe)
: base(new Reconciler(modelKind), input)
{
Input = input;
}
/// <include file='doc.xml' path='doc/members/member[@name="WordEmbeddings"]/*' />
/// <param name="input">Vector of tokenized text.</param>
/// <param name="modelKind">The pretrained word embedding model.</param>
/// <returns></returns>
public static Vector <float> WordEmbeddings(this VarVector <string> input, WordEmbeddingsTransform.PretrainedModelKind modelKind = WordEmbeddingsTransform.PretrainedModelKind.Sswe)
{
Contracts.CheckValue(input, nameof(input));
return(new OutColumn(input, modelKind));
}
public WordEmbeddingsEstimator(IHostEnvironment env, string inputColumn, string outputColumn,
WordEmbeddingsTransform.PretrainedModelKind modelKind = WordEmbeddingsTransform.PretrainedModelKind.Sswe)
: this(env, modelKind, new WordEmbeddingsTransform.ColumnInfo(inputColumn, outputColumn))
{
}
