find-best-method

Search Wisent's steering methods for a benchmark and report the best-performing one, driving the sweep with a hyperparameter optimizer (Hyperopt or Optuna).

Basic Usage
python -m wisent find-best-method --model MODEL --task TASK --output-dir DIR --trials-multiplier N --backend {hyperopt,optuna}

Examples

Find the best method with Optuna
python -m wisent find-best-method   --model meta-llama/Llama-3.1-8B-Instruct   --task truthfulqa_mc1   --output-dir ./best_method   --trials-multiplier 10   --backend optuna

Arguments

Required

ArgumentDescription
--modelHuggingFace model ID
--taskBenchmark task name
--output-dirDirectory for results
--trials-multiplierTrials per dimension of the search space
--backendOptimizer backend: hyperopt or optuna

Related Commands

Stay in the loop. Never miss out.

Subscribe to our newsletter and unlock Wisent insights.