Skip to content

Commit

Permalink
Update outdated parts of ReadMe
Browse files Browse the repository at this point in the history
  • Loading branch information
ThomasMeissnerDS committed Oct 14, 2023
1 parent a2d97f4 commit 8ee2942
Showing 1 changed file with 1 addition and 3 deletions.
4 changes: 1 addition & 3 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -152,7 +152,7 @@ train-test-split, cross-validation can be enabled easily:

```sh
from bluecast.blueprints.cast import BlueCast
from bluecast.config.training_config import TrainingConfig, XgboostTuneParamsConfig
from bluecast.config.training_config import TrainingConfig


# Create a custom training config and adjust general training parameters
Expand Down Expand Up @@ -234,7 +234,6 @@ automl = BlueCast(
target_column="target"
conf_training=train_config,
conf_xgboost=xgboost_param_config,

)

automl.fit(df_train, target_col="target")
Expand Down Expand Up @@ -392,7 +391,6 @@ y_probs, y_classes = automl.predict(df_val)
Also this step can be customized. The following example shows how to:
```sh
from bluecast.config.training_config import FeatureSelectionConfig
from bluecast.config.training_config import TrainingConfig
from bluecast.preprocessing.custom import CustomPreprocessing
from sklearn.feature_selection import RFECV
Expand Down

0 comments on commit 8ee2942

Please sign in to comment.