Машинне навчання з великими наборами даних

Паралельне програмування з Dask у Python

James Fulton

Climate Informatics Researcher

Завантаження й попередня обробка даних

# Завантаження табличного набору даних
import dask.dataframe as dd
dask_df = dd.read_parquet("dataset_parquet")

X = dask_df[['feature1', 'feature2', 'feature3']]
y = dask_df['target_column']

from dask_ml.preprocessing import StandardScaler

scaler = StandardScaler()


scaler.fit(X)  # Це не ліниво


standardized_X = scaler.transform(X)  # Це ліниво

Поділ на train/test

from dask_ml.model_selection import train_test_split

X_train, X_test, y_train, y_test = train_test_split(X, y, shuffle=True, test_size=0.2)

print(X_train)

Структура Dask DataFrame:
        feature1    feature2    feature3
npartitions=7                               
           int64     float64     float64
             ...         ...         ...

Оцінювання

# Перевірка навченої моделі на тренувальних даних
train_score = dask_model.score(X_train, y_train) # Не ліниво

print(train_score)

-0.12321

# Перевірка навченої моделі на тестових даних
test_score = dask_model.score(X_test, y_test)  # Не ліниво

print(test_score)

-0.23453

Давайте потренуємось!

Паралельне програмування з Dask у Python