API Reference
Documentation for the OptimalFeatureSelection
public interface.
Index
IAI.OptimalFeatureSelectionClassifier
IAI.OptimalFeatureSelectionLearner
IAI.OptimalFeatureSelectionRegressor
IAI.Questionnaire
IAI.get_prediction_constant
IAI.get_prediction_weights
IAI.show_questionnaire
IAI.variable_importance
IAI.write_questionnaire
Types
IAI.OptimalFeatureSelectionLearner
— TypeAbstract type encompassing all optimal feature selection learners.
IAI.OptimalFeatureSelectionClassifier
— TypeLearner for conducting optimal feature selection on classification problems.
The following parameters are supported (refer to the documentation for each):
IAI.OptimalFeatureSelectionRegressor
— TypeLearner for conducting optimal feature selection on regression problems.
The following parameters are supported (refer to the documentation for each):
Model Details
These functions can be used to query the structure of a OptimalFeatureSelectionLearner
. The examples make use of the following learner:
Fitted OptimalFeatureSelectionClassifier:
Constant: -3.01551
Weights:
region==C: 2.91716
score2: 0.0574938
(Higher score indicates stronger prediction for class `true`)
IAI.get_prediction_constant
— Functionget_prediction_constant(lnr::OptimalFeatureSelectionLearner)
Return the constant term in the prediction in the trained lnr
.
Example
IAI.get_prediction_constant(lnr)
-3.0155103
get_prediction_constant(lnr::GLMNetCVRegressor)
Return the constant term in the prediction made by the best fit on the path in lnr
.
get_prediction_constant(lnr::GLMNetCVRegressor, fit_index::Integer)
Return the constant term in the prediction made by the fit at fit_index
on the path in lnr
.
IAI.get_prediction_weights
— Functionget_prediction_weights(lnr::OptimalFeatureSelectionLearner)
Return the weights for each feature in the prediction in the trained lnr
. The weights are returned as two Dict
s, one for numeric features and one for categoric features.
The numeric Dict
has key-value pairs of feature names and their corresponding weights in the prediction.
The categoric Dict
has key-value pairs of feature names and a corresponding Dict
that maps the categoric levels for that feature to their weights in the prediction.
Any feature not included in either Dict
has zero weight in the prediction, and similarly, any categoric levels that are not included have zero weight.
Example
numeric_weights, categoric_weights = IAI.get_prediction_weights(lnr)
numeric_weights
Dict{Symbol,Float64} with 1 entry:
:score2 => 0.0574938
categoric_weights
Dict{Symbol,Dict{Any,Float64}} with 1 entry:
:region => Dict{Any,Float64}("C"=>2.91716)
get_prediction_weights(lnr::GLMNetCVRegressor)
Return the weights for each feature in the prediction made by the best fit on the path in lnr
.
get_prediction_weights(lnr::GLMNetCVRegressor, fit_index::Integer)
Return the weights for each feature in the prediction made by the fit at fit_index
on the path in lnr
.
Utilities
IAI.variable_importance
— Methodvariable_importance(lnr::OptimalFeatureSelectionLearner)
For OptimalFeatureSelectionLearner
s, the importance is determined using the coefficients in the resulting model after scaling to account for features of different magnitudes.
Visualization
Interactive Visualizations
IAI.write_questionnaire
— Methodwrite_questionnaire(f, lnr::OptimalFeatureSelectionLearner; keyword_arguments...)
write_questionnaire(f, grid::GridSearch; keyword_arguments...)
Write interactive questionnaire based on lnr
or grid
to f
in HTML format.
IAI.show_questionnaire
— Methodshow_questionnaire(lnr::OptimalFeatureSelectionLearner; keyword_arguments...)
show_questionnaire(grid::GridSearch; keyword_arguments...)
Show interactive questionnaire based on lnr
or grid
in default browser.
IAI.Questionnaire
— MethodQuestionnaire(lnr::OptimalFeatureSelectionLearner; keyword_arguments...)
Specifies an interactive questionnaire based on lnr
.