// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
// See the LICENSE file in the project root for more information.
 
namespace Microsoft.ML.Trainers
{
    /// <summary>
    /// Interface for the Trainer Estimator.
    /// </summary>
    /// <typeparam name="TTransformer">The type of the transformer returned by the estimator.</typeparam>
    /// <typeparam name="TModel">The type of the model parameters.</typeparam>
    public interface ITrainerEstimator<out TTransformer, out TModel> : IEstimator<TTransformer>
        where TTransformer : IPredictionTransformer<TModel>
        where TModel : class
    {
        /// <summary>
        /// Gets the <see cref="TrainerInfo"/> information about the trainer.
        /// </summary>
        TrainerInfo Info { get; }
    }
}