UBC-MDS / DSCI_522_Alberta-Oil-Spills

1 stars 2 forks source link

Feedback from classmates #27

Closed alyciakb closed 5 years ago

alyciakb commented 5 years ago
    depth_range = range(1,10)
    train_cv = []
    test_cv = []
    for d in depth_range:
        model = DecisionTreeClassifier(max_depth=d)
        train_cv.append(np.mean(cross_val_score(model, X_train, y_train, cv=10)))
        --test_cv.append(np.mean(cross_val_score(model, X_test, y_test, cv=10)))-- REMOVE
    max_cv = max(train_cv)
    opt_d = train_cv.index(max_cv)

Feedback from Tony

alyciakb commented 5 years ago

✅ Cross Validation code updated in script 3

alyciakb commented 5 years ago