List of all items

Structs

• bench::classification_metrics::ClassificationMetrics
• bench::classification_profiler::ClassificationProfiler
• bench::core::train_metrics::TrainMetrics
• bench::regression_metrics::RegressionMetrics
• bench::regression_profiler::RegressionProfiler
• builders::linear_regression::LinearRegressionBuilder
• builders::logistic_regression::LogisticRegressionBuilder
• core::activations::leaky_relu::LeakyReLU
• core::activations::relu::ReLU
• core::activations::sigmoid::Sigmoid
• core::activations::tanh::Tanh
• model::linear_regression::LinearRegression
• model::logistic_regression::LogisticRegression
• optim::sgd::optimizer::GradientDescent

Enums

• bench::core::error::ProfilerError
• core::activations::activation_functions::ActivationFn
• core::error::ModelError

Traits

• bench::core::profiler::Profiler
• builders::builder::Builder
• core::activations::activation::Activation
• model::core::base::BaseModel
• model::core::base::OptimizableModel
• model::core::classification_model::ClassificationModel
• model::core::param_collection::GradientCollection
• model::core::param_collection::ParamCollection
• model::core::regression_model::RegressionModel
• optim::core::optimizer::Optimizer
• optim::core::state::OptimizerState

Functions

• utils::data::get_features_and_target
• utils::data::load_dataset
• utils::data::shuffle_split

Type Aliases

• core::types::Matrix
• core::types::ParamsHashmap
• core::types::Scalar
• core::types::Vector