chore: reorganize ai module

rust-dd · Oct 1, 2024 · 5c851ed · 5c851ed
1 parent cb8a1c6
commit 5c851ed
Show file tree

Hide file tree

Showing 7 changed files with 9 additions and 6 deletions.
diff --git a/Cargo.toml b/Cargo.toml
@@ -18,6 +18,7 @@ approx = "0.5.1"
 candle-core = "0.7.2"
 candle-datasets = "0.7.2"
 candle-nn = "0.7.2"
+candle-transformers = "0.7.2"
 chrono = "0.4.38"
 indicatif = "0.17.8"
 levenberg-marquardt = "0.14.0"
@@ -45,6 +46,7 @@ time = { version = "0.3.36", features = [
     "parsing",
 ], optional = true }
 tokio-test = "0.4.4"
+tracing = "0.1.40"
 yahoo_finance_api = { version = "*", git = "https://github.com/dancixx/yahoo_finance_api.git", rev = "476ebdc", optional = true }
 
 [dev-dependencies]

diff --git a/src/ai.rs b/src/ai.rs
@@ -1 +1 @@
-pub mod fou_lstm;
+pub mod fou;
diff --git a/src/ai/fou.rs b/src/ai/fou.rs
@@ -0,0 +1,4 @@
+pub mod fou_lstm_datasets;
+pub mod fou_lstm_model_1_d;
+pub mod fou_lstm_model_2_d;
+pub mod fou_vae;
diff --git a/src/ai/fou_lstm/datasets.rs → src/ai/fou/fou_lstm_datasets.rs b/src/ai/fou_lstm/datasets.rs → src/ai/fou/fou_lstm_datasets.rs
diff --git a/src/ai/fou_lstm/lstm_model_1_d.rs → src/ai/fou/fou_lstm_model_1_d.rs b/src/ai/fou_lstm/lstm_model_1_d.rs → src/ai/fou/fou_lstm_model_1_d.rs
@@ -8,7 +8,7 @@ use candle_nn::{
 };
 use polars::prelude::*;
 
-use super::datasets::test_vasicek_1_d;
+use super::fou_lstm_datasets::test_vasicek_1_d;
 
 pub struct Model {
   is_train: bool,

diff --git a/src/ai/fou_lstm/lstm_model_2_d.rs → src/ai/fou/fou_lstm_model_2_d.rs b/src/ai/fou_lstm/lstm_model_2_d.rs → src/ai/fou/fou_lstm_model_2_d.rs
@@ -7,7 +7,7 @@ use candle_nn::{
 };
 use polars::prelude::*;
 
-use super::datasets::test_vasicek_2_d;
+use super::fou_lstm_datasets::test_vasicek_2_d;
 
 pub struct Model {
   is_train: bool,

diff --git a/src/ai/fou_lstm.rs b/src/ai/fou_lstm.rs