diff --git a/ml/src/asset_selection/mod.rs b/ml/src/asset_selection/mod.rs index 0ebb3c352..d3438bd4a 100644 --- a/ml/src/asset_selection/mod.rs +++ b/ml/src/asset_selection/mod.rs @@ -8,6 +8,9 @@ use serde::{Deserialize, Serialize}; pub mod scorer; pub mod selector; +pub use scorer::{AssetScore, PredictabilityScorer, PredictionResult, ScoringWeights}; +pub use selector::{ActiveSetConfig, ActiveSetSelector, TradingCandidate}; + /// Classification of a tradable asset. #[derive(Debug, Clone, PartialEq, Eq, Hash, Serialize, Deserialize)] pub enum AssetClass { diff --git a/ml/src/data_pipeline/mod.rs b/ml/src/data_pipeline/mod.rs index 09468c066..78fbcdd3f 100644 --- a/ml/src/data_pipeline/mod.rs +++ b/ml/src/data_pipeline/mod.rs @@ -8,6 +8,10 @@ pub mod cache; pub mod manager; pub mod prepared; +pub use cache::{CacheManifest, CachedRange, SymbolCache}; +pub use manager::{DataSplit, DatasetManager, FEATURE_DIM}; +pub use prepared::{Batch, BatchIterator, PreparedDataset}; + use chrono::NaiveDate; use serde::{Deserialize, Serialize}; diff --git a/ml/src/lib.rs b/ml/src/lib.rs index b569b98c1..08948b663 100644 --- a/ml/src/lib.rs +++ b/ml/src/lib.rs @@ -1962,4 +1962,10 @@ pub mod prelude { // Common external types pub use rust_decimal::Decimal; pub use serde::{Deserialize, Serialize}; + + // Data pipeline + pub use crate::data_pipeline::{DatasetManager, DatasetMode, DatasetSpec, PreparedDataset}; + + // Asset selection + pub use crate::asset_selection::{ActiveSetSelector, AssetUniverse, PredictabilityScorer}; }