hrntsm/Tunny

View on GitHub
Optuna/Sampler/BoTorchSampler.cs

Summary

Maintainability
A
0 mins
Test Coverage
namespace Optuna.Sampler
{
    /// <summary>
    /// https://optuna.readthedocs.io/en/stable/reference/generated/optuna.integration.BoTorchSampler.html
    /// </summary>
    public class BoTorchSampler : SamplerBase
    {
        public int NStartupTrials { get; set; } = 10;

        public dynamic ToPython(dynamic optuna, bool hasConstraints)
        {
            return optuna.integration.BoTorchSampler(
                seed: Seed,
                n_startup_trials: NStartupTrials,
                constraints_func: hasConstraints ? ConstraintFunc() : null
            );
        }
    }
}