ngboost_rs/
lib.rs

1pub mod api;
2pub mod dist;
3pub mod evaluation;
4pub mod learners;
5pub mod ngboost;
6pub mod scores;
7pub mod survival;
8
9// Re-export commonly used types at crate root
10pub use ngboost::{
11    EvalsResult, LearningRateSchedule, LineSearchMethod, NGBClassifier, NGBMultiClassifier,
12    NGBMultiClassifier10, NGBMultiClassifier3, NGBMultiClassifier4, NGBMultiClassifier5,
13    NGBRegressor, NGBoost, NGBoostParams,
14};
15pub use scores::{natural_gradient_regularized, CRPScore, LogScore, Scorable, Score};
16
17// Re-export distribution traits
18pub use dist::{ClassificationDistn, Distribution, DistributionMethods, RegressionDistn};
19
20// Re-export evaluation functions
21pub use evaluation::{
22    brier_score, calculate_calib_error, calibration_curve_data, calibration_regression,
23    calibration_time_to_event, concordance_index, concordance_index_uncensored_only, log_loss,
24    mean_absolute_error, mean_squared_error, pit_histogram, root_mean_squared_error,
25    CalibrationCurveData, CalibrationResult, PITHistogramData,
26};