Update scripts/download_data.py
Browse files- scripts/download_data.py +1 -1
scripts/download_data.py
CHANGED
|
@@ -5,7 +5,7 @@ from sklearn.model_selection import train_test_split
|
|
| 5 |
|
| 6 |
def download_data():
|
| 7 |
# Загрузка датасета diabetes с помощью fetch_openml
|
| 8 |
-
cardio_data = fetch_openml("Cardiovascular-Disease-dataset", version=1, parser="auto")
|
| 9 |
cardio_data_df = cardio_data.frame
|
| 10 |
cardio_data_df['cardio'] = cardio_data_df['cardio'].apply(lambda x: 'positive' if x=='1' else 'negative')
|
| 11 |
|
|
|
|
| 5 |
|
| 6 |
def download_data():
|
| 7 |
# Загрузка датасета diabetes с помощью fetch_openml
|
| 8 |
+
cardio_data = fetch_openml("Cardiovascular-Disease-dataset", version=1, parser="auto", data_home="/app/")
|
| 9 |
cardio_data_df = cardio_data.frame
|
| 10 |
cardio_data_df['cardio'] = cardio_data_df['cardio'].apply(lambda x: 'positive' if x=='1' else 'negative')
|
| 11 |
|