// Copyright (c) Microsoft Corporation. All rights reserved.
// Licensed under the MIT License.
// Code generated by Microsoft (R) AutoRest Code Generator.

package com.azure.resourcemanager.machinelearning.models;

import com.azure.core.annotation.Fluent;
import com.fasterxml.jackson.annotation.JsonProperty;
import java.time.Duration;
import java.util.List;

/** Forecasting Training related configuration. */
@Fluent
public final class ForecastingTrainingSettings extends TrainingSettings {
    /*
     * Allowed models for forecasting task.
     */
    @JsonProperty(value = "allowedTrainingAlgorithms")
    private List<ForecastingModels> allowedTrainingAlgorithms;

    /*
     * Blocked models for forecasting task.
     */
    @JsonProperty(value = "blockedTrainingAlgorithms")
    private List<ForecastingModels> blockedTrainingAlgorithms;

    /** Creates an instance of ForecastingTrainingSettings class. */
    public ForecastingTrainingSettings() {
    }

    /**
     * Get the allowedTrainingAlgorithms property: Allowed models for forecasting task.
     *
     * @return the allowedTrainingAlgorithms value.
     */
    public List<ForecastingModels> allowedTrainingAlgorithms() {
        return this.allowedTrainingAlgorithms;
    }

    /**
     * Set the allowedTrainingAlgorithms property: Allowed models for forecasting task.
     *
     * @param allowedTrainingAlgorithms the allowedTrainingAlgorithms value to set.
     * @return the ForecastingTrainingSettings object itself.
     */
    public ForecastingTrainingSettings withAllowedTrainingAlgorithms(
        List<ForecastingModels> allowedTrainingAlgorithms) {
        this.allowedTrainingAlgorithms = allowedTrainingAlgorithms;
        return this;
    }

    /**
     * Get the blockedTrainingAlgorithms property: Blocked models for forecasting task.
     *
     * @return the blockedTrainingAlgorithms value.
     */
    public List<ForecastingModels> blockedTrainingAlgorithms() {
        return this.blockedTrainingAlgorithms;
    }

    /**
     * Set the blockedTrainingAlgorithms property: Blocked models for forecasting task.
     *
     * @param blockedTrainingAlgorithms the blockedTrainingAlgorithms value to set.
     * @return the ForecastingTrainingSettings object itself.
     */
    public ForecastingTrainingSettings withBlockedTrainingAlgorithms(
        List<ForecastingModels> blockedTrainingAlgorithms) {
        this.blockedTrainingAlgorithms = blockedTrainingAlgorithms;
        return this;
    }

    /** {@inheritDoc} */
    @Override
    public ForecastingTrainingSettings withEnableDnnTraining(Boolean enableDnnTraining) {
        super.withEnableDnnTraining(enableDnnTraining);
        return this;
    }

    /** {@inheritDoc} */
    @Override
    public ForecastingTrainingSettings withEnableModelExplainability(Boolean enableModelExplainability) {
        super.withEnableModelExplainability(enableModelExplainability);
        return this;
    }

    /** {@inheritDoc} */
    @Override
    public ForecastingTrainingSettings withEnableOnnxCompatibleModels(Boolean enableOnnxCompatibleModels) {
        super.withEnableOnnxCompatibleModels(enableOnnxCompatibleModels);
        return this;
    }

    /** {@inheritDoc} */
    @Override
    public ForecastingTrainingSettings withEnableStackEnsemble(Boolean enableStackEnsemble) {
        super.withEnableStackEnsemble(enableStackEnsemble);
        return this;
    }

    /** {@inheritDoc} */
    @Override
    public ForecastingTrainingSettings withEnableVoteEnsemble(Boolean enableVoteEnsemble) {
        super.withEnableVoteEnsemble(enableVoteEnsemble);
        return this;
    }

    /** {@inheritDoc} */
    @Override
    public ForecastingTrainingSettings withEnsembleModelDownloadTimeout(Duration ensembleModelDownloadTimeout) {
        super.withEnsembleModelDownloadTimeout(ensembleModelDownloadTimeout);
        return this;
    }

    /** {@inheritDoc} */
    @Override
    public ForecastingTrainingSettings withStackEnsembleSettings(StackEnsembleSettings stackEnsembleSettings) {
        super.withStackEnsembleSettings(stackEnsembleSettings);
        return this;
    }

    /**
     * Validates the instance.
     *
     * @throws IllegalArgumentException thrown if the instance is not valid.
     */
    @Override
    public void validate() {
        super.validate();
    }
}
