def ext(X,y):
X_train,X_validation,y_train,y_validation = train_test_split(X,y,random_state=0)
ext = ExtraTreeRegressor(random_state=1)
ext.fit(X_train,y_train.ravel())
print 'training error:',1.0 - ext.score(X_train,y_train)
print 'validation error:',1.0 - ext.score(X_validation,y_validation)
time_fit(ext,X_train,y_train.ravel())
评论列表
文章目录