Description Usage Arguments Value
Tunes the hyperparameters of the desired algorithm/s using either hyperband or BOHB.
1 2 3 4 5 6 7 8 9 10 11 | autoRLearn_(
df_train,
df_test,
maxTime = 10,
models = c("randomForest", "naiveBayes", "boosting", "l2-linear-classifier", "svm"),
optimizationAlgorithm = "hyperband",
bw = 3,
kde_type = "single",
max_iter = 81,
metric = "acc"
)
|
df_train |
Dataframe of the training dataset. Assumes it is in perfect shape with all numeric variables and factor response variable named "class". |
df_test |
Dataframe of the test dataset. Assumes it is in perfect shape with all numeric variables and factor response variable named "class". |
maxTime |
Float representing the maximum time the algorithm should be run (in minutes). |
models |
List of strings denoting which algorithms to use for the process:
|
optimizationAlgorithm |
- String of which hyperparameter tuning algorithm to use:
|
bw |
- (only applies to BOHB) Double representing how much should the KDE bandwidth be widened. Higher values allow the algorithm to explore more hyperparameter combinations |
kde_type |
- (only applies to BOHB) String representing whether a model's hyperparameters should be tuned individually of each other or have their probability densities multiplied:
|
max_iter |
- (affects both hyperband and BOHB) Integer representing the maximum number of iterations that one successive halving run can have |
metric |
String of the evaluation metric to be used in the model performance optimization:
|
List of Results
perf
- Evaluated metric of the best performing model on the test data
pred
- prediction on the test data using the best model
model
- best model object
best_models
- table with the best hyperparameters found for the selected models.
Add the following code to your website.
For more information on customizing the embed code, read Embedding Snippets.