Skip to Content
ConfigurationParameter Reference

Parameter Reference

Complete reference for every parameter in Hedgehog’s YAML configuration files.


config.yml

The main configuration file. Controls input/output paths, parallelism, and references to all stage-specific configs.

ParameterTypeDefaultDescription
generated_mols_pathstringsrc/hedgehog/configs/examples/moses_1000.csvPath to the CSV file containing generated molecules
target_mols_pathstringsrc/hedgehog/configs/examples/target_mols.csvPath to the CSV file containing reference molecules
folder_to_savestringresults/runOutput directory where all pipeline results are saved
n_jobsint-1Number of parallel workers for CPU-bound tasks (-1 = all available cores. Prefer an explicit smaller number such as 4 or 8.)
sample_sizeint10000Number of molecules to sample from the input file (null = use all)
batch_sizeint512Batch size for descriptor computation and other batched operations
save_sampled_molsbooltrueWhether to save the sampled molecule subset to disk
large_dataset_modeboolfalseEnable streaming chunked processing for very large pre-docking dataset statistics
large_dataset_chunk_rowsint250000Rows per processing chunk in large dataset mode
large_dataset_single_csv_limitint1000000Maximum row count for also materializing compatibility CSV files from shard outputs
large_dataset_output_formatstringcsv.gzShard file format for large dataset row-level intermediate tables
large_dataset_filter_databoolfalseIn large dataset mode, whether filter pass/fail results should remove molecules from downstream outputs
large_dataset_enable_all_filtersbooltrueIn large dataset mode, enable configured descriptor/structural filters as calculations even when they do not filter outputs
pains_file_pathstringsrc/hedgehog/vendor/moleval/metrics/wehi_pains.csvPath to the PAINS filter definitions file
mcf_file_pathstringsrc/hedgehog/vendor/moleval/metrics/mcf.csvPath to the MCF definitions file
ligand_preparation_toolstring(proprietary path)Absolute path to an external ligand preparation binary
protein_preparation_toolstring(proprietary path)Absolute path to an external protein preparation binary
config_mol_prepstringsrc/hedgehog/configs/config_mol_prep.ymlPath to the preprocessing stage config
config_descriptorsstringsrc/hedgehog/configs/config_descriptors.ymlPath to the descriptors stage config
config_structFiltersstringsrc/hedgehog/configs/config_structFilters.ymlPath to the structural filters stage config
config_synthesisstringsrc/hedgehog/configs/config_synthesis.ymlPath to the synthesis stage config
config_dockingstringsrc/hedgehog/configs/config_docking.ymlPath to the docking stage config
config_docking_filtersstringsrc/hedgehog/configs/config_docking_filters.ymlPath to the docking filters stage config
config_weighted_scorestringsrc/hedgehog/configs/config_weighted_score.ymlPath to the weighted model assessment config
config_molevalstringsrc/hedgehog/configs/config_moleval.ymlPath to the MolEval reporting config

config_mol_prep.yml

Preprocessing. Standardizes molecules before any descriptor computation. This stage aims to produce “clean” molecules by:

  • removing salts and solvents and keeping the largest fragment
  • disconnecting metals
  • canonicalizing tautomers
  • removing stereochemistry
  • applying strict filters (allowed atom whitelist, no radicals, no isotopes, single fragment)

General Settings

ParameterTypeDefaultDescription
runbooltrueEnable or disable preprocessing
n_jobsint-1Worker count for molecule preparation
filters.allowed_atomslist[string][C, N, O, S, F, Cl, Br, I, P, H]Allowed atom symbols
filters.require_single_fragmentbooltrueReject multi-fragment molecules
filters.reject_radicalsbooltrueReject molecules with radical electrons
filters.reject_isotopesbooltrueReject isotopically labeled molecules
output.write_duplicates_removedbooltrueWrite duplicates_removed.csv when duplicates are dropped

config_descriptors.yml

Descriptors. Controls molecular descriptor calculation, filtering borders, and plotting options.

General Settings

The descriptors stage applies both layers together:

  • borders define generic descriptor ranges such as molWt, logP, TPSA, hbd, hba, n_rings, and fsp3.
  • structural_constraints are converted into additional upper bound checks on derived descriptor columns.

Use borders to shape broad property space and structural_constraints to cap specific motifs that can still pass those broad ranges.

Plotting Settings

ParameterTypeDescription
filtered_cols_to_plotlist[string]Descriptor columns to include in filtered distribution plots
discrete_features_to_plotlist[string]Columns treated as discrete (bar charts instead of KDE)
not_to_smooth_plot_by_sideslist[string]Columns where KDE side-smoothing is disabled
renamerdict[string, string]Display names for descriptors in plot labels

config_structFilters.yml

Structural Filters. Controls structural alert screening and medicinal chemistry filters. Molecules flagged by enabled filters are removed from the pipeline.

General Settings

ParameterTypeDefaultDescription
runbooltrueEnable or disable the structural filters stage
filter_databooltrueWhether to actually remove flagged molecules from downstream stages
parse_input_n_jobsint-1Worker count for parsing input molecules
write_per_filter_outputsbooltrueWrite per filter output folders and CSVs
generate_plotsbooltrueGenerate structural filter plots
generate_failure_analysisbooltrueGenerate failure-analysis outputs
combine_in_memorybooltrueCombine enabled filter results in memory before writing the final output
parallel_schedulerstringprocessesDefault scheduler for parallel filter execution

Structural Filter Profiles

The default structural filter configuration is the exploration profile in config_structFilters.yml. Three named ready-to-use profile files are shipped alongside it:

  • config_structFilters_strict.yml - conservative profile for high-confidence hygiene screening
  • config_structFilters_balanced.yml - practical mid-conservatism profile
  • config_structFilters_exploration.yml - least conservative profile for retaining more chemistry diversity

config_synthesis.yml

Synthesis Feasibility. Controls the retrosynthesis feasibility stage, including synthesizability score thresholds.

ParameterTypeDefaultDescription
runbooltrueEnable or disable the synthesis stage
n_jobsint-1Worker count for synthesis scoring and retrosynthesis (-1/0 = auto/all available cores)
enabled_scoreslistsa, syba, rascore, sync, scscore, nonpher, fsscore, gasaSynthesis score calculators to run. Optional scorers return NaN with warnings when their external dependencies are not configured
run_retrosynthesisbooltrueRun AiZynthFinder retrosynthetic analysis
filter_solved_onlybooltrueKeep only molecules for which a retrosynthetic route was found
sa_score_minfloat1Minimum synthetic accessibility score (Ertl)
sa_score_maxfloat4.5Maximum synthetic accessibility score (lower = easier to synthesize)
syba_score_minfloat0Minimum SYBA score (Bayesian synthesizability)
syba_score_maxfloatinfMaximum SYBA score
ra_score_minfloat0.5Minimum retrosynthetic accessibility score
ra_score_maxfloat1Maximum retrosynthetic accessibility score
sync_auto_installbooltrueDownload the SYNC checkpoint automatically when it is missing
sync_devicestringcpuTorch device for SYNC inference
sync_conformer_seedint61453RDKit ETKDG conformer seed for SYNC inputs
fsscore_pythonstring | nullnullPython interpreter for isolated FSScore worker environment
fsscore_model_pathstring | nullnullExplicit FSScore checkpoint path (*.ckpt)
fsscore_repo_pathstring | nullnullOptional FSScore checkout path used to resolve models/pretrain_graph_GGLGGL_ep242_best_valloss.ckpt
fsscore_batch_sizeint128Batch size passed to fsscore.score
fsscore_num_workersint | nullnullOptional dataloader worker count passed to fsscore.score
score_filtersobject{}Optional min/max filters for additional score columns such as sync_score, sc_score, nonpher_complexity_score, fs_score, or gasa_score
gasa.commandstringnullOptional local command template for batch gasa scoring using {input} and {output} placeholders
gasa.executablestringnullOptional local executable path/name used for gasa scoring (<exe> --smiles <SMILES>)
gasa.api_urlstringnullOptional local loopback HTTP endpoint for gasa scoring (POST {"smiles": ...})
gasa.timeout_secondsfloat30Timeout per gasa backend call

config_docking.yml

Docking. Controls molecular docking using SMINA, GNINA, Matcha, or any explicit combination of them. Defines the receptor, search box, and engine-specific parameters.

General Settings

ParameterTypeDefaultDescription
runbooltrueEnable or disable the docking stage
toolsstringgninaDocking engine selection: all, gnina, smina, matcha, or a comma-separated list such as gnina,matcha
receptor_pdbstringsrc/hedgehog/configs/examples/7EW9_apo.pdbPath to the receptor PDB file
auto_runbooltrueAutomatically start docking after ligand preparation
run_in_backgroundboolfalseRun docking as a background process
prepare_ligandsboolfalseUse external ligand preparation before docking. false keeps the input molecule mapping as close to 1:1 as possible; true may expand one input molecule into multiple prepared ligands
gnina_per_process_cpuintgnina_config.cpuCPU threads per GNINA process in per molecule mode
gnina_parallel_jobs_maxint6Upper bound for auto GNINA per molecule job count

SMINA Configuration (smina_config)

ParameterTypeDefaultDescription
binstringsminaPath or name of the SMINA binary (resolved via PATH if not absolute)
autobox_ligandstringsrc/hedgehog/configs/examples/05C_from_7EW9.sdfReference ligand SDF for automatic search box definition
autobox_addfloat4Padding (Angstroms) added to each side of the autobox
cpuint32Number of CPU threads for docking
seedint42Random seed for reproducibility
exhaustivenessint8Search exhaustiveness (higher = more thorough, slower)
num_modesint1Maximum number of binding modes to generate per ligand

GNINA Configuration (gnina_config)

ParameterTypeDefaultDescription
binstringgninaPath or name of the GNINA binary (resolved via PATH if not absolute)
autobox_ligandstringsrc/hedgehog/configs/examples/05C_from_7EW9.sdfReference ligand SDF for automatic search box definition
autobox_addfloat4Padding (Angstroms) added to each side of the autobox
cpuint8Number of CPU threads for docking
seedint42Random seed for reproducibility
no_gpuboolfalseDisable GPU acceleration (false keeps GPU enabled when available)
num_modesint1Maximum number of binding modes to generate per ligand

Matcha Configuration (matcha_config)

ParameterTypeDefaultDescription
checkout_dirstringmodules/matcha_remoteManaged Matcha checkout directory populated from GitHub
uv_binstringuvLauncher used to invoke Matcha
autobox_ligandstringsrc/hedgehog/configs/examples/05C_from_7EW9.sdfOptional Matcha autobox reference ligand
devicestringautoMatcha device selection (auto, cpu, cuda, cuda:N, mps)
n_samplesint20Number of Matcha poses generated per ligand
scorerstringgninaMatcha scorer mode (gnina, custom, none)
scorer_minimizebooltrueMinimize poses during Matcha GNINA scoring
physical_onlyboolfalseKeep only physically valid poses in Matcha outputs
keep_workdirboolfalsePreserve Matcha internal work directory after the run

When prepare_ligands is true, one input molecule may produce several prepared ligands. This can change row counts and downstream mapping. Keep it false for the default 1:1-oriented docking path unless you explicitly need an external preparation workflow.


config_docking_filters.yml

Three-Dimensional Filters. Post-docking 3D filters evaluate the quality of docked poses and remove poor candidates. Five independent filters can be combined with all (every filter must pass) or any (at least one must pass) aggregation.

General Settings

ParameterTypeDefaultDescription
runbooltrueEnable or disable the docking filters stage
run_after_dockingbooltrueAutomatically run after the docking stage completes
input_sdfstring | nullnullPath to input SDF; if null, uses docking output
receptor_pdbstring | nullnullPath to receptor PDB; if null, uses docking config value

Aggregation

ParameterTypeDefaultDescription
modestringallall = molecule must pass every enabled filter; any = pass at least one
save_metricsbooltrueSave detailed per molecule metrics to a CSV file
save_failedboolfalseSave molecules that failed filtering to a separate file

Deduplication

Docking can produce multiple poses per molecule when num_modes is greater than 1 (the default config uses num_modes: 1). After filtering, the pipeline deduplicates to unique molecules:

  1. All passing poses are saved to filtered_poses.csv (full pose-level detail)
  2. Poses are sorted by minimizedAffinity (best affinity first)
  3. For each unique mol_idx, only the best-scoring pose is kept
  4. Deduplicated molecules are saved to filtered_molecules.csv

SMILES for the output are taken from the original ligands.csv (2D SMILES) rather than regenerated from 3D coordinates, which preserves the original stereochemistry encoding.


config_weighted_score.yml

Controls the post-run Generator Reality Assessment used by HTML reporting and RUN_INFO.md.

The scorecard is explainable and intended to rank generator behavior, not to estimate hit probability. It also reports a secondary Final Candidate Pool Quality score for the survivor set.

General Settings

ParameterTypeDefaultDescription
runbooltrueEnable or disable weighted model scoring output
versionstringv1Internal scorecard schema version
modestringgenerator_realityScoring mode label for the gate-aware generator score
target_final_countint100Target final count retained for secondary candidate-pool yield scoring
target_final_retentionfloat0.10Target final retention rate for generator yield scoring
confidence.min_final_molecules_highint100Minimum final molecules for high confidence
confidence.min_final_molecules_mediumint30Minimum final molecules for medium confidence

Component Weights (weights)

ParameterTypeDefaultDescription
weights.yieldfloat0.30Weight for final retention against target
weights.physchemfloat0.15Weight for descriptor all pass gate survival
weights.structuralfloat0.25Weight for structural stage survival
weights.synthesisfloat0.10Weight for synthesis component
weights.docking_posefloat0.15Weight for docking/pipeline pose component
weights.diversityfloat0.05Weight for diversity metrics component

Weights are normalized over all configured components before scoring. When one component is unavailable, it is simply excluded, and the effective average is recomputed from the remaining available components.

physchem is measured from stages/02_descriptors_initial/filtered/pass_flags.csv as an all pass descriptor gate rate, so it reflects the early generated set rather than the final survivor pool. The mean flag pass rate is retained as evidence only. structural uses the stage survival rate from filtered and failed molecules, with the weakest structural filter as supporting evidence. Final descriptor files are used only as a fallback for older or partial runs. synthesis and docking_pose similarly prefer full stage evaluation artifacts before filtered or final survivor files.

Secondary Candidate Pool Weights (candidate_pool_weights)

candidate_pool_weights control the secondary Final Candidate Pool Quality score. It keeps the older survivor-pool interpretation: final-count yield saturation, mean descriptor flag pass rate, mean structural flag pass rate, and the same synthesis/docking/diversity formulas.

Yield and Structural Settings

ParameterTypeDefaultDescription
yield.modestringretentionUse final retention for the generator score; absolute restores count-saturation yield
yield.target_final_retentionfloat0.10Retention rate that maps to a full yield score
yield.count_weightfloat0.70Count-saturation weight for secondary candidate-pool yield
yield.retention_weightfloat0.30Log-retention weight for secondary candidate-pool yield
structural.stage_pass_weightfloat0.80Weight for structural stage survival
structural.worst_filter_weightfloat0.20Weight for the weakest structural filter pass rate

Hard Caps (hard_caps)

Hard caps prevent a model from receiving a high generator score when an early AND-gate rejects most molecules.

ParameterTypeDefaultDescription
hard_caps.structural_stage_pass_rate_belowfloat0.20Trigger threshold for structural stage survival
hard_caps.structural_stage_pass_rate_capfloat60.0Maximum score after structural cap trigger
hard_caps.descriptor_all_pass_rate_belowfloat0.50Trigger threshold for descriptor all pass survival
hard_caps.descriptor_all_pass_rate_capfloat70.0Maximum score after descriptor cap trigger
hard_caps.final_retention_rate_belowfloat0.05Trigger threshold for final retention
hard_caps.final_retention_rate_capfloat70.0Maximum score after retention cap trigger

Docking Thresholds (docking)

ParameterTypeDefaultDescription
docking.bad_affinityfloat-6.0Affinity at which docking contribution starts to approach zero
docking.good_affinityfloat-9.0Affinity at which docking affinity contribution reaches upper bound
docking.bad_cnnscorefloat0.35GNINA CNN score lower bound
docking.good_cnnscorefloat0.85GNINA CNN score upper bound
docking.bad_cnnaffinityfloat4.5CnnAffinity lower bound
docking.good_cnnaffinityfloat6.5CnnAffinity upper bound

Increase strictness by moving bad_* upward and good_* downward, or relax by widening the interval.

Synthesis Thresholds (synthesis)

ParameterTypeDefaultDescription
synthesis.sa_minfloat1.0Easier-to-synthesize SA floor
synthesis.sa_maxfloat4.5Harder-to-synthesize SA ceiling
synthesis.ra_minfloat0.5Minimum retrosynthetic accessibility minimum
synthesis.ra_maxfloat1.0Retrosynthetic accessibility maximum
synthesis.syba_midpointfloat0.0Sigmoid midpoint for SYBA
synthesis.syba_scalefloat50.0Sigmoid width for SYBA
synthesis.target_search_time_secfloat30.0Reference retrosynthesis search time
synthesis.search_time_scale_secfloat20.0Search-time penalty scale

Raise or lower these to bias toward faster/easier synthetic routes.


config_moleval.yml

Controls generative evaluation metrics computed during report generation. These metrics assess diversity, scaffold coverage, and basic filter pass rates across pipeline stages.

General Settings

ParameterTypeDefaultDescription
runbooltrueEnable or disable MolEval metric computation
n_jobsint-1Number of parallel workers for metric computation (-1 = all available cores)
devicestringcpuCompute device: cpu or cuda:0 (for neural metrics)
max_moleculesint2000Subsample threshold for O(N^2) metrics; datasets larger than this are subsampled

Metric Groups

Each flag enables or disables a group of related metrics.

ParameterTypeDefaultDescription
validityboolfalseCompute validity rate (disabled by default — always 1.0 after RDKit parsing)
uniquenessboolfalseCompute uniqueness rate (disabled by default — always 1.0 after deduplication)
internal_diversitybooltrueCompute IntDiv1 and IntDiv2 (intra-set Tanimoto diversity)
se_diversitybooltrueCompute sphere-exclusion diversity (SEDiv)
scaffold_diversitybooltrueCompute ScaffDiv and ScaffUniqueness (Murcko scaffold analysis)
functional_groupsbooltrueCompute functional group diversity ratio (FG)
ring_systemsbooltrueCompute ring system diversity ratio (RS)
filtersbooltrueCompute MCF + PAINS filter passage rate
mce18booltrueCompute mean MCE-18 molecular complexity score
Last updated on