Add 02 more tabular datasets

ludwig/datasets/allstate_claims_severity/__init__.py

@@ -0,0 +1,51 @@
+#! /usr/bin/env python
+# coding=utf-8
+# Copyright (c) 2021 Uber Technologies, Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+# ==============================================================================
+import os
+import pandas as pd
+
+from ludwig.datasets.base_dataset import BaseDataset, DEFAULT_CACHE_LOCATION
+from ludwig.datasets.mixins.kaggle import KaggleDownloadMixin
+from ludwig.datasets.mixins.load import CSVLoadMixin
+from ludwig.datasets.mixins.process import IdentityProcessMixin
+
+
+def load(cache_dir=DEFAULT_CACHE_LOCATION, split=False,
+         kaggle_username=None, kaggle_key=None):
+    dataset = AllstateClaimsSeverity(
+        cache_dir=cache_dir,
+        kaggle_username=kaggle_username,
+        kaggle_key=kaggle_key
+    )
+    return dataset.load(split=split)
+
+
+class AllstateClaimsSeverity(CSVLoadMixin, IdentityProcessMixin, KaggleDownloadMixin, BaseDataset):
+    """Allstate Claims Severity
+
+    https://www.kaggle.com/c/allstate-claims-severity/overview
+    """
+
+    def __init__(
+            self,
+            cache_dir=DEFAULT_CACHE_LOCATION,
+            kaggle_username=None,
+            kaggle_key=None
+    ):
+        self.kaggle_username = kaggle_username
+        self.kaggle_key = kaggle_key
+        self.is_kaggle_competition = True
+        super().__init__(dataset_name='allstate_claim_severity', cache_dir=cache_dir)

ludwig/datasets/allstate_claims_severity/config.yaml

@@ -0,0 +1,7 @@
+version: 1.0
+competition: allstate-claims-severity
+archive_filename: allstate-claims-severity.zip
+split_filenames:
+  train_file: train.csv
+csv_filename: train.csv
+download_file_type: csv

ludwig/datasets/porto_seguro_safe_driver/__init__.py

@@ -0,0 +1,52 @@
+#! /usr/bin/env python
+# coding=utf-8
+# Copyright (c) 2021 Uber Technologies, Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+# ==============================================================================
+import os
+import pandas as pd
+
+from ludwig.datasets.base_dataset import BaseDataset, DEFAULT_CACHE_LOCATION
+from ludwig.datasets.mixins.kaggle import KaggleDownloadMixin
+from ludwig.datasets.mixins.load import CSVLoadMixin
+from ludwig.datasets.mixins.process import IdentityProcessMixin
+
+
+def load(cache_dir=DEFAULT_CACHE_LOCATION, split=False,
+         kaggle_username=None, kaggle_key=None):
+    dataset = PortoSeguroSafeDriver(
+        cache_dir=cache_dir,
+        kaggle_username=kaggle_username,
+        kaggle_key=kaggle_key
+    )
+    return dataset.load(split=split)
+
+
+class PortoSeguroSafeDriver(CSVLoadMixin, IdentityProcessMixin, KaggleDownloadMixin, BaseDataset):
+    """Porto Seguro's Safe Driver Prediction
+
+    https://www.kaggle.com/c/porto-seguro-safe-driver-prediction/overview
+    """
+
+    def __init__(
+            self,
+            cache_dir=DEFAULT_CACHE_LOCATION,
+            kaggle_username=None,
+            kaggle_key=None
+    ):
+        self.kaggle_username = kaggle_username
+        self.kaggle_key = kaggle_key
+        self.is_kaggle_competition = True
+        super().__init__(dataset_name='porto_seguro_safe_driver', cache_dir=cache_dir)
+

ludwig/datasets/porto_seguro_safe_driver/config.yaml

@@ -0,0 +1,7 @@
+version: 1.0
+competition: porto-seguro-safe-driver-prediction
+archive_filename: porto-seguro-safe-driver-prediction.zip
+split_filenames:
+  train_file: train.csv
+csv_filename: train.csv
+download_file_type: csv