Advertisement
DrSartoriuss

Untitled

Aug 22nd, 2023
61
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 1.72 KB | Software | 0 0
  1. #выбираем лучший алгоритм и максимальное число итераций для логистической регрессии
  2. best_model = None
  3. best_result = 0
  4. best_max_iter = 0
  5. best_solver = []
  6. solvers = ["lbfgs","newton-cg","liblinear"]
  7.  
  8. #задаем число максимальных итераций от 100 до 2000 с шагом 50
  9. #и алгоритм
  10. for max_iter in range(100, 2001, 50):
  11.     for solver in solvers:
  12.         #инициализируем модель
  13.         model = LogisticRegression(random_state=12345,
  14.                                    solver=solver, max_iter=max_iter)
  15.         #обучаем модель на тренировочной выборке
  16.         model.fit(data_train_features, data_train_target)
  17.         #получаем предсказания модели на валидационной выборке
  18.         predictions_valid = model.predict(data_val_features)
  19.         #считаем значение метрики accuracy на валидационной выборке
  20.         result_valid = accuracy_score(data_val_target, predictions_valid)
  21.         #выбираем лучшие параметры max_iter и solver
  22.         if result_valid > best_result:
  23.             best_model = model
  24.             best_result = result_valid
  25.             best_max_iter = max_iter
  26.             best_solver = solver            
  27. #выводим лучшие значения accuracy, количества итераций и алгоритма  
  28. print("Accuracy наилучшей модели на валидационной выборке:", best_result,
  29.           "Количество итераций:", max_iter, "Алгоритм:", best_solver)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement