File size: 1,022 Bytes
825e978 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 |
import os
import pandas as pd
import sklearn.datasets as skd
from pathlib import Path
def sklearnDatasets(url,query):
try:
current_directory = os.getcwd()
downloads_folder = os.path.join(current_directory, query)
print(f"{url} started downloading")
os.makedirs(downloads_folder, exist_ok=True)
dataset_func = getattr(skd, url, None)
if dataset_func:
dataset = dataset_func(data_home=downloads_folder)
X, y = dataset.data, dataset.target
df = pd.DataFrame(X, columns=dataset.feature_names)
df['Target'] = y
csv_file_path = os.path.join(downloads_folder, f"{url}.csv")
df.to_csv(csv_file_path, index=False)
# print(f"Dataset '{url}' downloaded and saved to {csv_file_path}")
else:
print(f"Unknown dataset function: {url}. Please check the URL.")
except Exception as e:
print(f"Dataset Not found")
# sklearnDatasets("fetch_california_housing")
|