File size: 751 Bytes
61f924c 3f5e3ec 61f924c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
# Загрузка необходимых библиотек
from sklearn.datasets import fetch_openml
from sklearn.model_selection import train_test_split
def download_data():
# Загрузка датасета diabetes с помощью fetch_openml
cardio_data = fetch_openml("Cardiovascular-Disease-dataset", version=1, parser="auto", data_home="/app/")
cardio_data_df = cardio_data.frame
cardio_data_df['cardio'] = cardio_data_df['cardio'].apply(lambda x: 'positive' if x=='1' else 'negative')
# Разделение данных на обучающую и тестовую выборки
train_set, test_set = train_test_split(cardio_data_df, test_size=0.1, random_state=42)
return train_set, test_set
|