dict comprehension

list comprehension만 자주 사용했었는데 요즘 dict comprehension으로 쉽게 풀리는 문제를 많이 만나게 되어 기억하기 위해 작성한다.

 

예)

DecisionTreeRegressor에서 최적의 max_leaf_nodes 구하는 코드

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)

candidate_max_leaf_nodes = [5, 25, 50, 100, 250, 500]
# Write loop to find the ideal tree size from candidate_max_leaf_nodes
scores = {leaf_size: get_mae(leaf_size, train_X, val_X, train_y, val_y) for leaf_size in candidate_max_leaf_nodes}
best_tree_size = min(scores, key=scores.get)

 

scores.get