Skip to content

Commit

Permalink
Add files via upload
Browse files Browse the repository at this point in the history
  • Loading branch information
pycaret authored Jan 25, 2020
1 parent ac55662 commit 476d95a
Show file tree
Hide file tree
Showing 4 changed files with 146 additions and 170 deletions.
8 changes: 4 additions & 4 deletions anomaly.py
Original file line number Diff line number Diff line change
Expand Up @@ -932,7 +932,7 @@ def tune_model(model=None,
Description:
------------
This function tunes the fraction parameter using a predefined grid with
This function tunes the fraction parameter using a predefined grid with
the objective of optimizing a supervised learning metric as defined in
the optimize param. You can choose the supervised estimator from a large
library available in pycaret. By default, supervised estimator is Linear.
Expand All @@ -945,7 +945,7 @@ def tune_model(model=None,
boston = get_data('boston')
experiment_name = setup(data = boston, normalize = True)
tuned_knn = tune_model(model = 'knn', supervised_target = 'medv', optimize = 'R2')
tuned_knn = tune_model(model = 'knn', supervised_target = 'medv')
This will return tuned k-Nearest Neighbors model.
Expand All @@ -971,7 +971,7 @@ def tune_model(model=None,
Stochastic Outlier Selection 'sos' module-pyod.models.sos
supervised_target: string
Name of target column for supervised learning. It cannot be None.
Name of the target column for supervised learning.
method: string, default = 'drop'
When method set to drop, it will drop the outlier rows from training dataset
Expand Down Expand Up @@ -1027,7 +1027,7 @@ def tune_model(model=None,
Light Gradient Boosting 'lightgbm' Regression
CatBoost Regressor 'catboost' Regression
If set to None, default is Linear model for both classification
If set to None, Linear model is used by default for both classification
and regression tasks.
optimize: string, default = None
Expand Down
6 changes: 3 additions & 3 deletions clustering.py
Original file line number Diff line number Diff line change
Expand Up @@ -915,7 +915,7 @@ def tune_model(model=None,
boston = get_data('boston')
experiment_name = setup(data = boston, normalize = True)
tuned_kmeans = tune_model(model = 'kmeans', supervised_target = 'medv', optimize = 'R2')
tuned_kmeans = tune_model(model = 'kmeans', supervised_target = 'medv')
This will return tuned K Means Clustering Model.
Expand All @@ -934,7 +934,7 @@ def tune_model(model=None,
K-Modes clustering 'kmodes' git/nicodv/kmodes
supervised_target: string
Name of target column for supervised learning.
Name of the target column for supervised learning.
estimator: string, default = None
Expand Down Expand Up @@ -984,7 +984,7 @@ def tune_model(model=None,
Light Gradient Boosting 'lightgbm' Regression
CatBoost Classifier 'catboost' Regression
If set to None, default is Linear model for both classification
If set to None, Linear model is used by default for both classification
and regression tasks.
optimize: string, default = None
Expand Down
Loading

0 comments on commit 476d95a

Please sign in to comment.