From a3b73b6f4f24d54b73890b795e12f42e2046a59d Mon Sep 17 00:00:00 2001 From: Jirka Date: Sat, 11 Nov 2023 13:10:50 +0100 Subject: [PATCH] docs: fix args in DataConfig --- docs/data.md | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/docs/data.md b/docs/data.md index ef93353b..359de1fc 100644 --- a/docs/data.md +++ b/docs/data.md @@ -5,16 +5,16 @@ Pytorch Tabular handles this using a `DataConfig` object. ## Basic Usage - `target`: List\[str\]: A list of strings with the names of the target column(s) -- `continuous_columns`: List\[str\]: Column names of the numeric fields. Defaults to \[\] -- `categorical_columns`: List\[str\]: Column names of the categorical fields to treat differently +- `continuous_cols`: List\[str\]: Column names of the numeric fields. Defaults to \[\] +- `categorical_cols`: List\[str\]: Column names of the categorical fields to treat differently ### Usage Example ```python data_config = DataConfig( target=["label"], - continuous_columns=["feature_1", "feature_2"], - categorical_columns=["cat_feature_1", "cat_feature_2"], + continuous_cols=["feature_1", "feature_2"], + categorical_cols=["cat_feature_1", "cat_feature_2"], ) ```