|
using System.Collections.Generic;
using Newtonsoft.Json;
using SweepableEstimator = Microsoft.ML.AutoML.SweepableEstimator;
using Microsoft.ML.AutoML.CodeGen;
using ColorsOrder = Microsoft.ML.Transforms.Image.ImagePixelExtractingEstimator.ColorsOrder;
using ColorBits = Microsoft.ML.Transforms.Image.ImagePixelExtractingEstimator.ColorBits;
using ResizingKind = Microsoft.ML.Transforms.Image.ImageResizingEstimator.ResizingKind;
using Anchor = Microsoft.ML.Transforms.Image.ImageResizingEstimator.Anchor;
using Microsoft.ML.SearchSpace;
namespace Microsoft.ML.AutoML.CodeGen
{
internal partial class NamedEntityRecognitionMulti : SweepableEstimator<NamedEntityRecognitionOption>
{
public NamedEntityRecognitionMulti(NamedEntityRecognitionOption defaultOption, SearchSpace<NamedEntityRecognitionOption> searchSpace = null)
{
this.TParameter = defaultOption;
this.SearchSpace = searchSpace;
this.EstimatorType = EstimatorType.NamedEntityRecognitionMulti;
}
internal NamedEntityRecognitionMulti()
{
this.EstimatorType = EstimatorType.NamedEntityRecognitionMulti;
this.TParameter = new NamedEntityRecognitionOption();
}
internal override IEnumerable<string> CSharpUsingStatements
{
get => new string[] {"using Microsoft.ML;", "using Microsoft.ML.Trainers;", "using Microsoft.ML.TorchSharp;", };
}
internal override IEnumerable<string> NugetDependencies
{
get => new string[] {"Microsoft.ML", "Microsoft.ML.TorchSharp", };
}
internal override string FunctionName
{
get => "MulticlassClassification.Trainers.NamedEntityRecognition";
}
}
}
|