96 lines
4.3 KiB
Python
96 lines
4.3 KiB
Python
from app.utils.system import clear_console
|
||
from app.zadachi.zadacha1 import zadacha1
|
||
from app.zadachi.zadacha2 import zadacha2
|
||
|
||
class menu:
|
||
|
||
def __init__(self):
|
||
pass # Конструктор пока пустой, при необходимости можно добавить параметры
|
||
|
||
def main_menu(self):
|
||
clear_console()
|
||
while True:
|
||
print('\nMain Menu:')
|
||
print('1. Задача 1')
|
||
print('2. Задача 2')
|
||
print('0. Exit')
|
||
|
||
choice = input('Select an option: ')
|
||
|
||
clear_console()
|
||
if choice == '1':
|
||
self.zadacha1_menu() # Переход к подменю Задачи 1
|
||
elif choice == '2':
|
||
self.zadacha2_menu() # Переход к подменю Задачи 2
|
||
elif choice == '0':
|
||
print('Exiting program...')
|
||
break
|
||
else:
|
||
print('Invalid option')
|
||
|
||
def zadacha1_menu(self):
|
||
"""Меню для задачи 1 с подкатегориями."""
|
||
clear_console()
|
||
while True:
|
||
print('\nЗадача 1:')
|
||
print('1. линейно')
|
||
print('2. квадратично')
|
||
print('0. Назад')
|
||
|
||
choice = input('Выберите подкатегорию: ')
|
||
|
||
clear_console()
|
||
z1 = zadacha1() # создаем объект для задачи 1
|
||
|
||
if choice == '1':
|
||
z1.plot_lin_data() # Выполнить первую подзадачу
|
||
#return # Вернуться в основное меню после выполнения
|
||
elif choice == '2':
|
||
z1.plot_sqr_data() # Выполнить вторую подзадачу
|
||
#return # Вернуться в основное меню после выполнения
|
||
elif choice == '0':
|
||
break # Вернуться в основное меню
|
||
else:
|
||
print('Неверный выбор, попробуйте снова.')
|
||
|
||
def zadacha2_menu(self):
|
||
"""Меню для задачи 2 (расчет коэффициентов методом наименьших квадратов)."""
|
||
clear_console()
|
||
z1 = zadacha1() # Используем объект задачи 1 для генерации данных
|
||
z2 = zadacha2() # Создаем объект для задачи 2
|
||
|
||
while True:
|
||
print('\nЗадача 2:')
|
||
print('1. Рассчитать коэффициенты b0 и b1 для линейной модели')
|
||
print('2. Сделать прогноз по рассчитанным коэффициентам')
|
||
print('0. Назад')
|
||
|
||
choice = input('Выберите подкатегорию: ')
|
||
|
||
clear_console()
|
||
|
||
if choice == '1':
|
||
# Генерация линейных данных с помощью функции из задачи 1
|
||
x_data, y_data = z1.create_lin_array()
|
||
|
||
# Рассчет коэффициентов b0 и b1 методом наименьших квадратов
|
||
b0, b1 = z2.calculate_coefficients(x_data, y_data)
|
||
|
||
print(f'Рассчитанные коэффициенты:')
|
||
print(f'b0 (свободный член) = {b0}')
|
||
print(f'b1 (коэффициент наклона) = {b1}')
|
||
elif choice == '2':
|
||
# Пример предсказания на новых данных
|
||
x_data, y_data = z1.create_lin_array() # Сначала генерируем данные
|
||
b0, b1 = z2.calculate_coefficients(x_data, y_data) # Рассчитываем коэффициенты
|
||
|
||
# Выполняем прогноз
|
||
predicted_y = z2.predict(x_data, b0, b1)
|
||
|
||
print(f'Предсказанные значения Y на основе рассчитанных коэффициентов:')
|
||
print(predicted_y)
|
||
elif choice == '0':
|
||
break # Вернуться в основное меню
|
||
else:
|
||
print('Неверный выбор, попробуйте снова.')
|