Advertisement
Guest User

Untitled

a guest
Feb 28th, 2020
170
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 1.53 KB | None | 0 0
  1. import pandas as pd
  2. from sklearn.model_selection import train_test_split
  3. from sklearn.ensemble import RandomForestRegressor
  4. from sklearn.metrics import r2_score
  5.  
  6. # прочитайте данные с атрибутами аккаунтов компаний на facebook и активностью на них
  7. fb = pd.read_csv('/datasets/dataset_facebook_cosmetics.csv', sep = ';')
  8.  
  9. # разделите данные на признаки (матрица X) и целевую переменную (y)
  10. X = fb.drop('Total Interactions', axis = 1)
  11. y = fb['Total Interactions']
  12.  
  13. # разделяем модель на обучающую и валидационную выборку
  14. X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2)
  15.  
  16. # зададим алгоритм для нашей модели
  17. model = RandomForestRegressor(solver='lbfgs')
  18.  
  19. # обучим модель
  20. model.fit(X_train, y_train)# обучите вашу модель на обучающей выборке
  21.  
  22. # воспользуемся уже обученной моделью, чтобы сделать прогнозы
  23. predictions = model.predict(X_test)# сделайте прогноз для валидационной выборки с помощью вашей модели
  24.  
  25. # оценим метрику R-квадрат на валидационной выборке и напечатаем
  26. r2 = r2_score(y_test, predictions)# напишите свой код здесь
  27. print('Значение метрики R-квадрат: ', r2)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement