library(optaas.client)

client <- OPTaaSClient$new("https://optaas.mindfoundry.ai", "Your OPTaaS API Key")

task <- client$create_task(
    title="Basic 2D Example",
    parameters=list(
        FloatParameter('x', minimum = -3, maximum = 1),
        FloatParameter('y', minimum = -6, maximum = 21)
    ),
    goal="min"
)

scoring_function <- function(x, y) {
    x**2 + y**2
}

best_result <- task$run(scoring_function, 15)

known_best <- list(x=0, y=0)
predictions <- task$get_surrogate_predictions(list(known_best))
print(paste("Score:", predictions[[1]]$mean, "Variance:", predictions[[1]]$variance))

far_away_points <- list(list(x=1, y=20), list(x=-3, y=-6))
predictions <- task$get_surrogate_predictions(far_away_points)
print(paste("Score:", predictions[[1]]$mean, "Variance:", predictions[[1]]$variance))
print(paste("Score:", predictions[[2]]$mean, "Variance:", predictions[[2]]$variance))