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 SentenceSimilarityRegression : SweepableEstimator<SentenceSimilarityOption>
    {
        public SentenceSimilarityRegression(SentenceSimilarityOption defaultOption, SearchSpace<SentenceSimilarityOption> searchSpace = null)
        {
            this.TParameter = defaultOption;
            this.SearchSpace = searchSpace;
            this.EstimatorType = EstimatorType.SentenceSimilarityRegression;
        }
 
        internal SentenceSimilarityRegression()
        {
            this.EstimatorType = EstimatorType.SentenceSimilarityRegression;
            this.TParameter = new SentenceSimilarityOption();
        }
    
        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 => "Regression.Trainers.SentenceSimilarity";
        }
    }
}