onnx-export-rs 0.1.1

Export canonical Rust machine-learning models to ONNX
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32

//! Source-library-independent inference representations.

mod affine_model;
mod categorical_model;
mod centroid_model;
mod dbscan_model;
mod gaussian_mixture;
mod glm_model;
mod knn_model;
mod linear_model;
mod logistic_model;
mod naive_bayes_model;
mod score_classifier;
mod svm_model;
mod tree_model;

pub use affine_model::AffineModel;
pub use categorical_model::CategoricalNaiveBayes;
pub use centroid_model::CentroidModel;
pub use dbscan_model::DbscanModel;
pub use gaussian_mixture::GaussianMixture;
pub use glm_model::{GeneralizedLinearModel, LinkFunction};
pub use knn_model::{KnnClassifier, KnnRegressor, KnnWeight};
pub use linear_model::LinearModelWeights;
pub use logistic_model::LogisticModelWeights;
pub use naive_bayes_model::GaussianNaiveBayes;
pub use score_classifier::LinearScoreClassifier;
pub use svm_model::{SvmClassifier, SvmKernel, SvmRegressor};
pub use tree_model::{
    flatten_tree, AggregationMode, BranchMode, ForestStructure, GradientBoostedEnsemble,
    PostTransform, RecursiveNode, TreeNode, TreeStructure, TreeTask,
};