gitadmin/foxhunt
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix(test): update mamba2 argmin tests for 14-param space (signal thresholds)

Browse Source 
Continuous vectors and expected counts updated from 12 to 14 params
after adding signal_high_bps and signal_low_bps to Mamba2Params.

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
jgrusewski
2026-03-11 16:12:14 +01:00
parent 43978e77a3
commit a4e9a80d6d
1 changed files with 11 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
crates/ml/src/hyperopt/tests_argmin.rs
View File

@@ -312,7 +312,7 @@ mod tests {
use crate::hyperopt::adapters::mamba2::Mamba2Params;
let bounds = Mamba2Params::continuous_bounds();
assert_eq!(bounds.len(), 12); // 12 parameters (total_decay_steps removed)
assert_eq!(bounds.len(), 14); // 12 base + signal_high_bps + signal_low_bps
// Learning rate (log scale)
assert!(bounds[0].0 < bounds[0].1);
@@ -344,7 +344,7 @@ mod tests {
assert!(result
.unwrap_err()
.to_string()
.contains("Expected 12 parameters"));
.contains("Expected 14 parameters"));
}
#[test]
@@ -352,7 +352,7 @@ mod tests {
use crate::hyperopt::adapters::mamba2::Mamba2Params;
let names = Mamba2Params::param_names();
assert_eq!(names.len(), 12); // 12 parameters (total_decay_steps removed)
assert_eq!(names.len(), 14); // 12 base + signal_high_bps + signal_low_bps
assert_eq!(names[0], "learning_rate");
assert_eq!(names[1], "batch_size");
assert_eq!(names[2], "dropout");
@@ -365,6 +365,8 @@ mod tests {
assert_eq!(names[9], "lookback_window");
assert_eq!(names[10], "sequence_stride");
assert_eq!(names[11], "norm_eps");
assert_eq!(names[12], "signal_high_bps");
assert_eq!(names[13], "signal_low_bps");
}
#[test]
@@ -385,6 +387,8 @@ mod tests {
60.0, // lookback_window
1.0, // sequence_stride
-11.5, // norm_eps (log scale)
10.0, // signal_high_bps
5.0, // signal_low_bps
];
let params = Mamba2Params::from_continuous(&continuous).unwrap();
@@ -409,6 +413,8 @@ mod tests {
60.0, // lookback_window
1.0, // sequence_stride
-11.5, // norm_eps (log scale)
10.0, // signal_high_bps
5.0, // signal_low_bps
];
let params = Mamba2Params::from_continuous(&continuous).unwrap();
@@ -427,6 +433,8 @@ mod tests {
60.0, // lookback_window
1.0, // sequence_stride
-11.5, // norm_eps (log scale)
10.0, // signal_high_bps
5.0, // signal_low_bps
];
let params2 = Mamba2Params::from_continuous(&continuous2).unwrap();