Advertisement
Guest User

Untitled

a guest
Feb 24th, 2020
2,870
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 1.55 KB | None | 0 0
  1. import pandas as pd
  2. from sklearn.tree import DecisionTreeClassifier
  3.  
  4. data = pd.read_csv('/datasets/heart.csv')
  5. features = data.drop(['target'], axis=1)
  6. target = data['target']
  7.  
  8. scores = []
  9.  
  10. # зададим размер блока, если их всего три
  11. block_size = int(len(data)/3)
  12.  
  13. for i in range(0, len(data), sample_size):
  14. # < запишите массив из индексов для валидационного блока >
  15. valid_indexes = data.iloc[i: i + block_size].index
  16. # < запишите массив из индексов для обучающей выборки >
  17. train_indexes = (data.iloc[:i] + data.iloc[i + block_size:]).index
  18.  
  19. # разбейте переменные features и target на выборки features_train, target_train, features_valid, target_valid
  20. # < напишите код здесь >
  21. features_train = features.iloc[train_indexes]
  22. features_valid = features.iloc[valid_indexes]
  23. target_train = target.iloc[train_indexes]
  24. target_valid = target.iloc[valid_indexes]
  25.  
  26. model = DecisionTreeClassifier(random_state=0)
  27. model = model.fit(features_train, target_train)
  28. model_predictions = model.predict(features_train)
  29. # < оцените качество модели >
  30. score = f1_score(target_train, model_predictions)
  31.  
  32. scores.append(score)
  33.  
  34. final_score = score.mean()
  35.  
  36. # < посчитайте среднее качество модели >
  37. print('Средняя оценка качества модели:', final_score)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement