fix: modules

rust-dd · Nov 9, 2024 · 83f2cbf · 83f2cbf
1 parent 2ded411
commit 83f2cbf
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 6 deletions.
diff --git a/src/ai/volatility/heston.rs b/src/ai/volatility/heston.rs
@@ -247,11 +247,11 @@ mod tests {
     let model = train(
       dataset,
       &Device::Cpu,
-      5,  // input_dim (parameters)
-      30, // hidden_size
-      88, // output_dim (implied volatilities)
-      32, // batch_size
-      5,  // epochs
+      5,   // input_dim (parameters)
+      30,  // hidden_size
+      88,  // output_dim (implied volatilities)
+      32,  // batch_size
+      200, // epochs
     )?;
 
     // Sample index for plotting

diff --git a/src/stats.rs b/src/stats.rs
@@ -2,6 +2,5 @@ pub mod cir;
 pub mod double_exp;
 pub mod fd;
 pub mod fou_estimator;
-pub mod fou_estimator_v2;
 pub mod mle;
 pub mod non_central_chi_squared;