"""Utilities to load popular datasets and artificial data generators.""" # Authors: The scikit-learn developers # SPDX-License-Identifier: BSD-3-Clause import textwrap from sklearn.datasets._base import ( clear_data_home, fetch_file, get_data_home, load_breast_cancer, load_diabetes, load_digits, load_files, load_iris, load_linnerud, load_sample_image, load_sample_images, load_wine, ) from sklearn.datasets._california_housing import fetch_california_housing from sklearn.datasets._covtype import fetch_covtype from sklearn.datasets._kddcup99 import fetch_kddcup99 from sklearn.datasets._lfw import fetch_lfw_pairs, fetch_lfw_people from sklearn.datasets._olivetti_faces import fetch_olivetti_faces from sklearn.datasets._openml import fetch_openml from sklearn.datasets._rcv1 import fetch_rcv1 from sklearn.datasets._samples_generator import ( make_biclusters, make_blobs, make_checkerboard, make_circles, make_classification, make_friedman1, make_friedman2, make_friedman3, make_gaussian_quantiles, make_hastie_10_2, make_low_rank_matrix, make_moons, make_multilabel_classification, make_regression, make_s_curve, make_sparse_coded_signal, make_sparse_spd_matrix, make_sparse_uncorrelated, make_spd_matrix, make_swiss_roll, ) from sklearn.datasets._species_distributions import fetch_species_distributions from sklearn.datasets._svmlight_format_io import ( dump_svmlight_file, load_svmlight_file, load_svmlight_files, ) from sklearn.datasets._twenty_newsgroups import ( fetch_20newsgroups, fetch_20newsgroups_vectorized, ) __all__ = [ "clear_data_home", "dump_svmlight_file", "fetch_20newsgroups", "fetch_20newsgroups_vectorized", "fetch_california_housing", "fetch_covtype", "fetch_file", "fetch_kddcup99", "fetch_lfw_pairs", "fetch_lfw_people", "fetch_olivetti_faces", "fetch_openml", "fetch_rcv1", "fetch_species_distributions", "get_data_home", "load_breast_cancer", "load_diabetes", "load_digits", "load_files", "load_iris", "load_linnerud", "load_sample_image", "load_sample_images", "load_svmlight_file", "load_svmlight_files", "load_wine", "make_biclusters", "make_blobs", "make_checkerboard", "make_circles", "make_classification", "make_friedman1", "make_friedman2", "make_friedman3", "make_gaussian_quantiles", "make_hastie_10_2", "make_low_rank_matrix", "make_moons", "make_multilabel_classification", "make_regression", "make_s_curve", "make_sparse_coded_signal", "make_sparse_spd_matrix", "make_sparse_uncorrelated", "make_spd_matrix", "make_swiss_roll", ] def __getattr__(name): if name == "load_boston": msg = textwrap.dedent( """ `load_boston` has been removed from scikit-learn since version 1.2. The Boston housing prices dataset has an ethical problem: as investigated in [1], the authors of this dataset engineered a non-invertible variable "B" assuming that racial self-segregation had a positive impact on house prices [2]. Furthermore the goal of the research that led to the creation of this dataset was to study the impact of air quality but it did not give adequate demonstration of the validity of this assumption. The scikit-learn maintainers therefore strongly discourage the use of this dataset unless the purpose of the code is to study and educate about ethical issues in data science and machine learning. In this special case, you can fetch the dataset from the original source:: import pandas as pd import numpy as np data_url = "http://lib.stat.cmu.edu/datasets/boston" raw_df = pd.read_csv(data_url, sep="\\s+", skiprows=22, header=None) data = np.hstack([raw_df.values[::2, :], raw_df.values[1::2, :2]]) target = raw_df.values[1::2, 2] Alternative datasets include the California housing dataset and the Ames housing dataset. You can load the datasets as follows:: from sklearn.datasets import fetch_california_housing housing = fetch_california_housing() for the California housing dataset and:: from sklearn.datasets import fetch_openml housing = fetch_openml(name="house_prices", as_frame=True) for the Ames housing dataset. [1] M Carlisle. "Racist data destruction?" [2] Harrison Jr, David, and Daniel L. Rubinfeld. "Hedonic housing prices and the demand for clean air." Journal of environmental economics and management 5.1 (1978): 81-102. """ ) raise ImportError(msg) try: return globals()[name] except KeyError: # This is turned into the appropriate ImportError raise AttributeError