Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Dataengines2 #49

Merged
merged 12 commits into from
Feb 26, 2024
Merged

Dataengines2 #49

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
12 commits
Select commit Hold shift + click to select a range
d049e6a
data engine 3
tvdboom Feb 11, 2024
8885b7a
set_output
tvdboom Feb 13, 2024
8a1a8a8
dataengines 4
tvdboom Feb 14, 2024
9562c49
dataengines 5
tvdboom Feb 15, 2024
2e59415
dataengines 5
tvdboom Feb 18, 2024
1e99a1c
dataengines 6
tvdboom Feb 18, 2024
19588ce
fix mypy in utils
tvdboom Feb 20, 2024
043aa7d
mypy fix basemodel
MarcovdBoom Feb 20, 2024
3a51c67
fix partial tests
tvdboom Feb 23, 2024
63ac60d
fixing type hints
tvdboom Feb 25, 2024
c3d9c6b
fixing type hints 2
tvdboom Feb 25, 2024
a59a3b5
dataengines final
MarcovdBoom Feb 26, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
26 changes: 21 additions & 5 deletions atom/_show_versions.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -20,12 +20,11 @@
"atom",
"beartype",
"category_encoders",
"dagshub",
"dill",
"featuretools",
"gplearn",
"imblearn",
"ipywidgets",
"featuretools",
"joblib",
"matplotlib",
"mlflow",
Expand All @@ -35,14 +34,31 @@
"optuna",
"pandas",
"plotly",
"ray",
"requests",
"sklearn",
"sklearnex", # Has no __version__ attribute
"scipy",
"shap",
"sktime",
"statsmodels",
"zoofs", # Has no __version__ attribute
"botorch",
"catboost",
"dagshub",
"dask[distributed]",
"explainerdashboard",
"gradio",
"lightgbm",
"modin[ray]",
"polars",
"pyarrow",
"pyspark",
"ray[serve]",
"requests",
"sklearnex",
"schemdraw",
"statsforecast",
"sweetviz",
"wordcloud",
"xgboost",
]


Expand Down
80 changes: 52 additions & 28 deletions atom/api.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -158,20 +158,20 @@ class ATOMClassifier(ATOM):
**X, train, test: dataframe-like**<br>
Feature set with shape=(n_samples, n_features).

**y: int, str or sequence**<br>
Target column corresponding to `X`.
**y: int, str, sequence or dataframe-like**<br>
Target column(s) corresponding to `X`.

- If int: Position of the target column in X.
- If str: Name of the target column in X.
- If int: Position of the target column in `X`.
- If str: Name of the target column in `X`.
- If sequence: Target column with shape=(n_samples,) or
sequence of column names or positions for multioutput tasks.
- If dataframe: Target columns for multioutput tasks.
- If dataframe-like: Target columns for multioutput tasks.

y: int, str, dict, sequence or dataframe, default=-1
Target column corresponding to `X`.
y: int, str, sequence or dataframe-like, default=-1
Target column(s) corresponding to `X`.

- If int: Position of the target column in X.
- If str: Name of the target column in X.
- If int: Position of the target column in `X`.
- If str: Name of the target column in `X`.
- If sequence: Target column with shape=(n_samples,) or
sequence of column names or positions for multioutput tasks.
- If dataframe: Target columns for multioutput tasks.
Expand Down Expand Up @@ -257,9 +257,16 @@ class ATOMClassifier(ATOM):

- "data":

- "numpy"
- "pandas" (default)
- "pandas-pyarrow"
- "polars"
- "polars-lazy"
- "pyarrow"
- "modin"
- "dask"
- "pyspark"
- "pyspark-pandas"

- "estimator":

Expand All @@ -276,6 +283,7 @@ class ATOMClassifier(ATOM):
parallelism. Less robust than `loky`.
- "threading": Single-node, thread-based parallelism.
- "ray": Multi-node, process-based parallelism.
- "dask": Multi-node, process-based parallelism.

memory: bool, str, Path or Memory, default=False
Enables caching for memory optimization. Read more in the
Expand Down Expand Up @@ -428,24 +436,24 @@ class ATOMForecaster(ATOM):
Exogenous feature set corresponding to y, with shape=(n_samples,
n_features).

**y: int, str or sequence**<br>
**y: int, str, sequence or dataframe-like**<br>
Time series.

- If int: Position of the target column in X.
- If str: Name of the target column in X.
- If int: Position of the target column in `X`.
- If str: Name of the target column in `X`.
- If sequence: Target column with shape=(n_samples,) or
sequence of column names or positions for multioutput tasks.
- If dataframe: Target columns for multioutput tasks.
- If dataframe-like: Target columns for multioutput tasks.

y: int, str, dict, sequence or dataframe, default=-1
y: int, str, sequence or dataframe-like, default=-1
Time series.

- If None: y is ignored.
- If int: Position of the target column in X.
- If str: Name of the target column in X.
- If None: `y` is ignored.
- If int: Position of the target column in `X`.
- If str: Name of the target column in `X`.
- If sequence: Target column with shape=(n_samples,) or
sequence of column names or positions for multioutput tasks.
- If dataframe: Target columns for multioutput tasks.
- If dataframe-like: Target columns for multioutput tasks.

This parameter is ignored if the time series is provided
through `arrays`.
Expand Down Expand Up @@ -526,9 +534,16 @@ class ATOMForecaster(ATOM):

- "data":

- "numpy"
- "pandas" (default)
- "pandas-pyarrow"
- "polars"
- "polars-lazy"
- "pyarrow"
- "modin"
- "dask"
- "pyspark"
- "pyspark-pandas"

- "estimator":

Expand All @@ -545,6 +560,7 @@ class ATOMForecaster(ATOM):
parallelism. Less robust than `loky`.
- "threading": Single-node, thread-based parallelism.
- "ray": Multi-node, process-based parallelism.
- "dask": Multi-node, process-based parallelism.

memory: bool, str, Path or Memory, default=False
Enables caching for memory optimization. Read more in the
Expand Down Expand Up @@ -689,24 +705,24 @@ class ATOMRegressor(ATOM):
**X, train, test: dataframe-like**<br>
Feature set with shape=(n_samples, n_features).

**y: int, str or sequence**<br>
Target column corresponding to `X`.
**y: int, str, sequence or dataframe-like**<br>
Target column(s) corresponding to `X`.

- If int: Position of the target column in X.
- If str: Name of the target column in X.
- If int: Position of the target column in `X`.
- If str: Name of the target column in `X`.
- If sequence: Target column with shape=(n_samples,) or
sequence of column names or positions for multioutput tasks.
- If dataframe: Target columns for multioutput tasks.

y: int, str, dict, sequence or dataframe, default=-1
Target column corresponding to `X`.
y: int, str, sequence or dataframe-like, default=-1
Target column(s) corresponding to `X`.

- If None: y is ignored.
- If int: Position of the target column in X.
- If str: Name of the target column in X.
- If None: `y` is ignored.
- If int: Position of the target column in `X`.
- If str: Name of the target column in `X`.
- If sequence: Target column with shape=(n_samples,) or
sequence of column names or positions for multioutput tasks.
- If dataframe: Target columns for multioutput tasks.
- If dataframe-like: Target columns for multioutput tasks.

This parameter is ignored if the target column is provided
through `arrays`.
Expand Down Expand Up @@ -775,9 +791,16 @@ class ATOMRegressor(ATOM):

- "data":

- "numpy"
- "pandas" (default)
- "pandas-pyarrow"
- "polars"
- "polars-lazy"
- "pyarrow"
- "modin"
- "dask"
- "pyspark"
- "pyspark-pandas"

- "estimator":

Expand All @@ -794,6 +817,7 @@ class ATOMRegressor(ATOM):
parallelism. Less robust than `loky`.
- "threading": Single-node, thread-based parallelism.
- "ray": Multi-node, process-based parallelism.
- "dask": Multi-node, process-based parallelism.

memory: bool, str, Path or Memory, default=False
Enables caching for memory optimization. Read more in the
Expand Down
Loading
Loading