fix: modules

Author: dancixx

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

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;