What the Forest Leaned On

10 pts · 10 coins

Machine Learning in CodeMedium

What the Forest Leaned On

A winery wants to know which chemical measurements matter most for telling its three grape varieties apart.

Write top_wine_features(n).

Load the data with load_wine(return_X_y=True, as_frame=True) and split it with train_test_split(X, y, test_size=0.25, random_state=0, stratify=y).
Fit RandomForestClassifier(n_estimators=100, random_state=0) on the training rows.
Rank the columns by feature_importances_, biggest first.

Return a list of the n most important column names.

Examples

n = 1

returns ['proline']

n = 3

returns ['proline', 'flavanoids', 'color_intensity']

Hints

solution.py

1import pandas as pd
2from sklearn.datasets import load_wine
3from sklearn.ensemble import RandomForestClassifier
4from sklearn.model_selection import train_test_split
5
6
7def top_wine_features(n):
8    pass
9