Skip to content

Instantly share code, notes, and snippets.

@pprett
pprett / test_subset.py
Created August 7, 2012 10:07
Exhibits error in GradientBoostingClassifier
from numpy import genfromtxt
from sklearn.ensemble import GradientBoostingClassifier
def main():
dataset = genfromtxt(open('train_subset.csv','r'), delimiter=',', dtype='float64')
clf = GradientBoostingClassifier(n_estimators=100, learn_rate=1.0, max_depth=1, random_state=0)
X = dataset[:,1:]
y = dataset[:,0]