Add OpenML 5 datasets

dunnkers · Jun 5, 2021 · 008f9ed · 008f9ed
1 parent 73ea17c
commit 008f9ed
Show file tree

Hide file tree

Showing 7 changed files with 30 additions and 11 deletions.
diff --git a/README.md b/README.md
@@ -16,7 +16,7 @@ fseval --help
 
 Now, create a [wandb](https://wandb.ai/) account and login to the CLI. We are now able to run benchmarks 💪🏻. The results will automatically be uploaded to the wandb dashboard.
 
-Run ReliefF on Iris dataset:
+Run ANOVA F-Value on Iris dataset:
 ```shell
 fseval dataset=iris estimator@pipeline.ranker=anova_f_value
 ```

diff --git a/fseval/conf/dataset/dresses_sales.yaml b/fseval/conf/dataset/dresses_sales.yaml
diff --git a/fseval/conf/dataset/madelon.yaml b/fseval/conf/dataset/madelon.yaml
@@ -0,0 +1,7 @@
+name: Madelon
+task: classification
+domain: synthetic
+adapter:
+  _target_: fseval.adapters.OpenML
+  dataset_id: 1485
+  target_column: Class
diff --git a/fseval/conf/dataset/nomao.yaml b/fseval/conf/dataset/nomao.yaml
@@ -0,0 +1,7 @@
+name: Nomao
+task: classification
+domain: geodata
+adapter:
+  _target_: fseval.adapters.OpenML
+  dataset_id: 1486
+  target_column: Class
diff --git a/fseval/conf/dataset/ozone_levels.yaml b/fseval/conf/dataset/ozone_levels.yaml
@@ -0,0 +1,7 @@
+name: Ozone Levels
+task: classification
+domain: nature
+adapter:
+  _target_: fseval.adapters.OpenML
+  dataset_id: 1487
+  target_column: Class
diff --git a/fseval/conf/dataset/phoneme.yaml b/fseval/conf/dataset/phoneme.yaml
@@ -0,0 +1,7 @@
+name: Phoneme
+task: classification
+domain: biomedical
+adapter:
+  _target_: fseval.adapters.OpenML
+  dataset_id: 1489
+  target_column: Class
diff --git a/setup.py b/setup.py
@@ -13,7 +13,7 @@
         long_description_content_type="text/markdown",
         license="MIT",
         author="Jeroen Overschie",
-        url="https://dunnkers.com/",
+        url="https://github.com/dunnkers/fseval",
         include_package_data=True,
         install_requires=[
             "hydra-core==1.1.0.rc1",