Skip to content
Merged
Show file tree
Hide file tree
Changes from 8 commits
Commits
Show all changes
29 commits
Select commit Hold shift + click to select a range
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
60 changes: 40 additions & 20 deletions src/Microsoft.ML.Transforms/EntryPoints/TextAnalytics.cs
Original file line number Diff line number Diff line change
Expand Up @@ -17,9 +17,9 @@ namespace Microsoft.ML.Runtime.Transforms
/// </summary>
public static class TextAnalytics
{
[TlcModule.EntryPoint(Name = "Transforms.TextFeaturizer",
Desc = Data.TextTransform.Summary,
UserName = Data.TextTransform.UserName,
[TlcModule.EntryPoint(Name = "Transforms.TextFeaturizer",
Desc = Data.TextTransform.Summary,
UserName = Data.TextTransform.UserName,
ShortName = Data.TextTransform.LoaderSignature,
XmlInclude = new[] { @"<include file='../Microsoft.ML.Transforms/Text/doc.xml' path='doc/members/member[@name=""TextTransform""]/*' />" ,
@"<include file='../Microsoft.ML.Transforms/Text/doc.xml' path='doc/members/example[@name=""TextTransform""]/*' />"})]
Expand All @@ -34,9 +34,9 @@ public static CommonOutputs.TransformOutput TextTransform(IHostEnvironment env,
};
}

[TlcModule.EntryPoint(Name = "Transforms.WordTokenizer",
[TlcModule.EntryPoint(Name = "Transforms.WordTokenizer",
Desc = Data.DelimitedTokenizeTransform.Summary,
UserName = Data.DelimitedTokenizeTransform.UserName,
UserName = Data.DelimitedTokenizeTransform.UserName,
ShortName = Data.DelimitedTokenizeTransform.LoaderSignature,
XmlInclude = new[] { @"<include file='../Microsoft.ML.Transforms/Text/doc.xml' path='doc/members/member[@name=""WordTokenizer""]/*' />",
@"<include file='../Microsoft.ML.Transforms/Text/doc.xml' path='doc/members/example[@name=""WordTokenizer""]/*' />"})]
Expand All @@ -51,9 +51,9 @@ public static CommonOutputs.TransformOutput DelimitedTokenizeTransform(IHostEnvi
};
}

[TlcModule.EntryPoint(Name = "Transforms.NGramTranslator",
Desc = NgramTransform.Summary,
UserName = NgramTransform.UserName,
[TlcModule.EntryPoint(Name = "Transforms.NGramTranslator",
Desc = NgramTransform.Summary,
UserName = NgramTransform.UserName,
ShortName = NgramTransform.LoaderSignature,
XmlInclude = new[] { @"<include file='../Microsoft.ML.Transforms/Text/doc.xml' path='doc/members/member[@name=""NgramTranslator""]/*' />" })]
public static CommonOutputs.TransformOutput NGramTransform(IHostEnvironment env, NgramTransform.Arguments input)
Expand All @@ -67,9 +67,9 @@ public static CommonOutputs.TransformOutput NGramTransform(IHostEnvironment env,
};
}

[TlcModule.EntryPoint(Name = "Transforms.Dictionarizer",
Desc = Data.TermTransform.Summary,
UserName = Data.TermTransform.UserName,
[TlcModule.EntryPoint(Name = "Transforms.Dictionarizer",
Desc = Data.TermTransform.Summary,
UserName = Data.TermTransform.UserName,
ShortName = Data.TermTransform.LoaderSignature)]
public static CommonOutputs.TransformOutput TermTransform(IHostEnvironment env, TermTransform.Arguments input)
{
Expand All @@ -82,9 +82,9 @@ public static CommonOutputs.TransformOutput TermTransform(IHostEnvironment env,
};
}

[TlcModule.EntryPoint(Name = "Transforms.SentimentAnalyzer",
Desc = "Uses a pretrained sentiment model to score input strings",
UserName = SentimentAnalyzingTransform.UserName,
[TlcModule.EntryPoint(Name = "Transforms.SentimentAnalyzer",
Desc = "Uses a pretrained sentiment model to score input strings",
UserName = SentimentAnalyzingTransform.UserName,
ShortName = SentimentAnalyzingTransform.ShortName,
XmlInclude = new[] { @"<include file='../Microsoft.ML.Transforms/Text/doc.xml' path='doc/members/member[@name=""SentimentAnalyzer""]/*' />",
@"<include file='../Microsoft.ML.Transforms/Text/doc.xml' path='doc/members/example[@name=""SentimentAnalyzer""]/*' />"})]
Expand All @@ -99,9 +99,9 @@ public static CommonOutputs.TransformOutput AnalyzeSentiment(IHostEnvironment en
};
}

[TlcModule.EntryPoint(Name = "Transforms.CharacterTokenizer",
Desc = CharTokenizeTransform.Summary,
UserName = CharTokenizeTransform.UserName,
[TlcModule.EntryPoint(Name = "Transforms.CharacterTokenizer",
Desc = CharTokenizeTransform.Summary,
UserName = CharTokenizeTransform.UserName,
ShortName = CharTokenizeTransform.LoaderSignature,
XmlInclude = new[] { @"<include file='../Microsoft.ML.Transforms/Text/doc.xml' path='doc/members/member[@name=""CharacterTokenizer""]/*' />" })]
public static CommonOutputs.TransformOutput CharTokenize(IHostEnvironment env, CharTokenizeTransform.Arguments input)
Expand All @@ -118,9 +118,9 @@ public static CommonOutputs.TransformOutput CharTokenize(IHostEnvironment env, C
};
}

[TlcModule.EntryPoint(Name = "Transforms.LightLda",
Desc = LdaTransform.Summary,
UserName = LdaTransform.UserName,
[TlcModule.EntryPoint(Name = "Transforms.LightLda",
Desc = LdaTransform.Summary,
UserName = LdaTransform.UserName,
ShortName = LdaTransform.ShortName,
XmlInclude = new[] { @"<include file='../Microsoft.ML.Transforms/Text/doc.xml' path='doc/members/member[@name=""LightLDA""]/*' />",
@"<include file='../Microsoft.ML.Transforms/Text/doc.xml' path='doc/members/example[@name=""LightLDA""]/*' />" })]
Expand All @@ -137,5 +137,25 @@ public static CommonOutputs.TransformOutput LightLda(IHostEnvironment env, LdaTr
OutputData = view
};
}

[TlcModule.EntryPoint(Name = "Transforms.WordEmbeddings",
Desc = WordEmbeddingsTransform.Summary,
UserName = WordEmbeddingsTransform.UserName,
ShortName = WordEmbeddingsTransform.ShortName,
XmlInclude = new[] { @"<include file='../Microsoft.ML.Transforms/Text/doc.xml' path='doc/members/member[@name=""WordEmbeddings""]/*' />",
@"<include file='../Microsoft.ML.Transforms/Text/doc.xml' path='doc/members/example[@name=""WordEmbeddings""]/*' />" })]
public static CommonOutputs.TransformOutput WordEmbeddings(IHostEnvironment env, WordEmbeddingsTransform.Arguments input)
{
Contracts.CheckValue(env, nameof(env));
env.CheckValue(input, nameof(input));

var h = EntryPointUtils.CheckArgsAndCreateHost(env, "WordEmbeddings", input);
var view = new WordEmbeddingsTransform(h, input, input.Data);
return new CommonOutputs.TransformOutput()
{
Model = new TransformModel(h, view, input.Data),
OutputData = view
};
}
}
}
Loading