Advertisement
Guest User

Untitled

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