← На главную Воспоминания

Суббота, 23 Сентября 2023

Начал курс по машин лернингу на кагле. Изучил до андерфиттинга и оверфитинга. Ссылка на курс. Кратко распишу что изучил

Пример кода на дефайн модели, ее фит и расчет средней обсолютной ошибки

from sklearn.metrics import mean_absolute_error
from sklearn.tree import DecisionTreeRegressor

def get_mae(max_leaf_nodes, train_X, val_X, train_y, val_y):
    model = DecisionTreeRegressor(max_leaf_nodes=max_leaf_nodes, random_state=0)
    model.fit(train_X, train_y)
    preds_val = model.predict(val_X)
    mae = mean_absolute_error(val_y, preds_val)
    return(mae)