Added TabularRegressionData extending TabularData (Lightning-Universe…

…#574) * added TabularClassificationData,TabularRegressionData extending TabularData * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * Update flash/tabular/regression/data.py Co-authored-by: thomas chaton <[email protected]> * Update flash/tabular/classification/data.py Co-authored-by: thomas chaton <[email protected]> * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * added TabularClassificationData,TabularRegressionData extending TabularData * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * PEP8 fix * modified tests * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com> Co-authored-by: thomas chaton <[email protected]>
sumanmichael · Jul 13, 2021 · c318e4a · c318e4a
1 parent f7a86ea
commit c318e4a
Show file tree

Hide file tree

Showing 11 changed files with 553 additions and 520 deletions.
diff --git a/README.md b/README.md
@@ -260,13 +260,13 @@ To illustrate, say we want to build a model to predict if a passenger survived o
 from torchmetrics.classification import Accuracy, Precision, Recall
 import flash
 from flash.core.data.utils import download_data
-from flash.tabular import TabularClassifier, TabularData
+from flash.tabular import TabularClassifier, TabularClassificationData
 
 # 1. Download the data
 download_data("https://pl-flash-data.s3.amazonaws.com/titanic.zip", 'data/')
 
 # 2. Load the data
-datamodule = TabularData.from_csv(
+datamodule = TabularClassificationData.from_csv(
     ["Sex", "Age", "SibSp", "Parch", "Ticket", "Cabin", "Embarked"],
     "Fare",
     target_fields="Survived",

diff --git a/flash/tabular/__init__.py b/flash/tabular/__init__.py
@@ -1 +1,3 @@
-from flash.tabular.classification import TabularClassifier, TabularData  # noqa: F401
+from flash.tabular.classification import TabularClassificationData, TabularClassifier  # noqa: F401
+from flash.tabular.data import TabularData  # noqa: F401
+from flash.tabular.regression import TabularRegressionData  # noqa: F401
diff --git a/flash/tabular/classification/__init__.py b/flash/tabular/classification/__init__.py
@@ -1,2 +1,2 @@
-from flash.tabular.classification.data import TabularData  # noqa: F401
+from flash.tabular.classification.data import TabularClassificationData  # noqa: F401
 from flash.tabular.classification.model import TabularClassifier  # noqa: F401