Skip to content

Conversation

Sohambasu07
Copy link
Collaborator

Features and/or fixes:

Code for Reproducibility:

import numpy as np

import neps
from neps import AskAndTell, algorithms
from neps.space.parsing import convert_to_space


def evaluate_pipeline(float1, float2, float3, float4, cat, float5, float6):
    return -float(
        np.sum([float1, float2, float3, float4, float5, float6])
    )

pipeline_space = {
    "float1": neps.Float(lower=0, upper=1, prior=1, prior_confidence="high"),
    "float2": neps.Float(lower=75, upper=1000, prior=85, prior_confidence="high"),
    "cat": neps.Categorical(choices=["a", "b", "c"]),
    "float3": neps.Float(lower=1, upper=100, is_fidelity=True),
    "float4": neps.Float(lower=4, upper=6),
    "float5": neps.Float(lower=0, upper=10, prior=5, prior_confidence="high"),
    "float6": neps.Float(lower=1, upper=3),
}

optimizer = AskAndTell(
    algorithms.PredefinedOptimizers["priorband"](
        space=convert_to_space(pipeline_space),
    )
)

def set_seed(seed: int):  # noqa: D103
    import random

    import numpy as np
    import torch

    np.random.seed(seed)
    random.seed(seed)
    torch.manual_seed(seed)

set_seed(0)
for i in range(200):
    print(f"Iteration {i + 1}")
    trial = optimizer.ask()
    x = trial.config
    y = evaluate_pipeline(**x)
    optimizer.tell(trial=trial, result=y)

@Sohambasu07 Sohambasu07 added the enhancement New feature or request label May 26, 2025
@Sohambasu07 Sohambasu07 merged commit 3dc7cb8 into master May 26, 2025
14 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

enhancement New feature or request

Projects

Status: Done

Development

Successfully merging this pull request may close these issues.

1 participant