Skip to content
This repository has been archived by the owner on Aug 1, 2024. It is now read-only.

Commit

Permalink
add ames housing
Browse files Browse the repository at this point in the history
  • Loading branch information
jeromedockes committed Apr 26, 2024
1 parent 4b684d6 commit ed8e16d
Showing 1 changed file with 8 additions and 5 deletions.
13 changes: 8 additions & 5 deletions doc/make_doc.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,7 @@
from pathlib import Path

import polars as pl
import pandas as pd
from skrub import datasets as skrub_data
from sklearn import datasets as sklearn_data
from skrubview import Report
Expand All @@ -18,16 +19,17 @@
reports_dir = build_dir / "reports"
reports_dir.mkdir()


AMES_HOUSING_CSV = "https://www.openml.org/data/get_csv/20649135/file2ed11cebe25.arff"
datasets = [(pd.read_csv(AMES_HOUSING_CSV), "AMES Housing")]
skrub_dataset_names = [
"employee_salaries",
"medical_charge",
"traffic_violations",
"drug_directory"
]
datasets = [
(getattr(skrub_data, f"fetch_{name}")().X, name) for name in skrub_dataset_names
"drug_directory",
]
datasets.extend(
[(getattr(skrub_data, f"fetch_{name}")().X, name) for name in skrub_dataset_names]
)

sklearn_dataset_names = ["titanic"]
datasets.extend(
Expand All @@ -42,6 +44,7 @@
]
)


def add_report(df, name):
print(f"making report for {name}", end="", flush=True)
df = pl.from_pandas(df)
Expand Down

0 comments on commit ed8e16d

Please sign in to comment.