Guest User

Untitled

a guest
Jan 23rd, 2018
81
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 1.13 KB | None | 0 0
  1. import numpy as np
  2. import pandas as pd
  3. from sklearn.model_selection import cross_val_score
  4. from sklearn.cross_validation import train_test_split
  5. from sklearn.ensemble import AdaBoostClassifier
  6.  
  7.  
  8.  
  9. data = pd.read_csv(r'D:downloadENB2012.csv', ';')
  10. data.head()
  11. kfold = 5 #количество подвыборок для валидации
  12. itog_val = {} #список для записи результатов кросс валидации разных алгоритмов
  13. X = data.drop('Y1', axis=1).values[:, :7] # [y] столбец не должен попадать в [X]
  14. y = data['Y1'].values
  15. X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2)
  16. print ('обучающая выборка:n', X_train[:9])
  17. print ('n')
  18. print ('тестовая выборка:n', X_test[:7])
  19. clf = AdaBoostClassifier(n_estimators=70)
  20. scores = cross_val_score(clf, X_train, y_train, cv=kfold)
  21. itog_val['AdaBoostClassifier'] = scores.mean()
  22. print ('итог', itog_val)
  23. clf.fit(X_train, y_train)
  24. clf.score(X_test, y_test)
  25. y_test_predicted = clf.predict(X_test)
  26. print ('AdaBoostClassifier first 9 predicted values:n', y_test_predicted[:9])
Add Comment
Please, Sign In to add comment