Fixing typo in ExtractWordEmbeddings (#1573)
* Fixing typo in ExtractWordEmbeddings * Fixing an API sample
This commit is contained in:
Родитель
b888db2897
Коммит
ac71ae4d1d
|
@ -104,7 +104,7 @@ namespace Microsoft.ML
|
|||
/// <param name="inputColumn">The input column.</param>
|
||||
/// <param name="outputColumn">The optional output column. If it is <value>null</value> the input column will be substituted with its value.</param>
|
||||
/// <param name="modelKind">The embeddings <see cref="WordEmbeddingsTransform.PretrainedModelKind"/> to use. </param>
|
||||
public static WordEmbeddingsExtractorEstimator ExtractWordEmbeedings(this TransformsCatalog.TextTransforms catalog,
|
||||
public static WordEmbeddingsExtractorEstimator ExtractWordEmbeddings(this TransformsCatalog.TextTransforms catalog,
|
||||
string inputColumn,
|
||||
string outputColumn = null,
|
||||
WordEmbeddingsTransform.PretrainedModelKind modelKind = WordEmbeddingsTransform.PretrainedModelKind.Sswe)
|
||||
|
@ -117,7 +117,7 @@ namespace Microsoft.ML
|
|||
/// <param name="inputColumn">The input column.</param>
|
||||
/// <param name="outputColumn">The optional output column. If it is <value>null</value> the input column will be substituted with its value.</param>
|
||||
/// <param name="customModelFile">The path of the pre-trained embeedings model to use. </param>
|
||||
public static WordEmbeddingsExtractorEstimator ExtractWordEmbeedings(this TransformsCatalog.TextTransforms catalog,
|
||||
public static WordEmbeddingsExtractorEstimator ExtractWordEmbeddings(this TransformsCatalog.TextTransforms catalog,
|
||||
string inputColumn,
|
||||
string customModelFile,
|
||||
string outputColumn = null)
|
||||
|
@ -130,7 +130,7 @@ namespace Microsoft.ML
|
|||
/// <param name="catalog">The text-related transform's catalog.</param>
|
||||
/// <param name="modelKind">The embeddings <see cref="WordEmbeddingsTransform.PretrainedModelKind"/> to use. </param>
|
||||
/// <param name="columns">The array columns, and per-column configurations to extract embeedings from.</param>
|
||||
public static WordEmbeddingsExtractorEstimator ExtractWordEmbeedings(this TransformsCatalog.TextTransforms catalog,
|
||||
public static WordEmbeddingsExtractorEstimator ExtractWordEmbeddings(this TransformsCatalog.TextTransforms catalog,
|
||||
WordEmbeddingsTransform.PretrainedModelKind modelKind = WordEmbeddingsTransform.PretrainedModelKind.Sswe,
|
||||
params WordEmbeddingsTransform.ColumnInfo[] columns)
|
||||
=> new WordEmbeddingsExtractorEstimator(Contracts.CheckRef(catalog, nameof(catalog)).GetEnvironment(), modelKind, columns);
|
||||
|
|
|
@ -321,7 +321,7 @@ namespace Microsoft.ML.Tests.Scenarios.Api.CookbookSamples
|
|||
|
||||
// NLP pipeline 4: word embeddings.
|
||||
.Append(mlContext.Transforms.Text.TokenizeWords("NormalizedMessage", "TokenizedMessage"))
|
||||
.Append(mlContext.Transforms.Text.ExtractWordEmbeedings("TokenizedMessage", "Embeddings",
|
||||
.Append(mlContext.Transforms.Text.ExtractWordEmbeddings("TokenizedMessage", "Embeddings",
|
||||
WordEmbeddingsTransform.PretrainedModelKind.GloVeTwitter25D));
|
||||
|
||||
// Let's train our pipeline, and then apply it to the same data.
|
||||
|
|
Загрузка…
Ссылка в новой задаче