Fitting decision tree on wrong data frame - cross validation intro section

[yap-alan]

Hi, I believe there is an error in the code to fit a decision tree on the red wine quality dataset.

/# train the model on the provided features
/# and mapped quality from before
clf.fit(df_train[cols], df_test.quality)

Shouldn't the feature column be from df_train as opposed to from df_test. This error was not repeated in the longer code section where you plot accuracies for different values of max_depth

Cheers

[abhishekkrthakur]

are you reading the version downloaded from somewhere else than this repo?