|
74 | 74 | //! use automl::settings::{
|
75 | 75 | //! Algorithm, CategoricalNBParameters, DecisionTreeClassifierParameters, Distance,
|
76 | 76 | //! GaussianNBParameters, KNNAlgorithmName, KNNClassifierParameters, KNNWeightFunction, Kernel,
|
77 |
| -//! LogisticRegressionParameters, Metric, RandomForestClassifierParameters, SVCParameters, |
| 77 | +//! LogisticRegressionParameters, LogisticRegressionSolverName, Metric, |
| 78 | +//! RandomForestClassifierParameters, SVCParameters, |
78 | 79 | //! };
|
79 | 80 | //!
|
80 | 81 | //! let settings = automl::Settings::default_classification()
|
|
91 | 92 | //! .with_n_trees(100)
|
92 | 93 | //! .with_min_samples_split(20),
|
93 | 94 | //! )
|
94 |
| -//! .with_logistic_settings(LogisticRegressionParameters::default()) |
| 95 | +//! .with_logistic_settings( |
| 96 | +//! LogisticRegressionParameters::default() |
| 97 | +//! .with_alpha(1.0) |
| 98 | +//! .with_solver(LogisticRegressionSolverName::LBFGS), |
| 99 | +//! ) |
95 | 100 | //! .with_svc_settings(
|
96 | 101 | //! SVCParameters::default()
|
97 | 102 | //! .with_epoch(10)
|
@@ -160,6 +165,9 @@ pub use smartcore::ensemble::random_forest_classifier::RandomForestClassifierPar
|
160 | 165 | /// Parameters for logistic regression (re-export from [Smartcore](https://docs.rs/smartcore/))
|
161 | 166 | pub use smartcore::linear::logistic_regression::LogisticRegressionParameters;
|
162 | 167 |
|
| 168 | +/// Parameters for logistic regression (re-export from [Smartcore](https://docs.rs/smartcore/)) |
| 169 | +pub use smartcore::linear::logistic_regression::LogisticRegressionSolverName; |
| 170 | + |
163 | 171 | /// Parameters for decision tree classification (re-export from [Smartcore](https://docs.rs/smartcore/))
|
164 | 172 | pub use smartcore::tree::decision_tree_classifier::DecisionTreeClassifierParameters;
|
165 | 173 |
|
|
0 commit comments