forked from UNN/2026-rff_mp
[2] task2
This commit is contained in:
MariiaOs
parent
cf755242e1
commit
3428678d8a
283
osipovamd/maze_project/experiment.py
Normal file
283
osipovamd/maze_project/experiment.py
Normal file
|
|
@ -0,0 +1,283 @@
|
|||
"""
|
||||
Экспериментальный запуск для всех лабиринтов и алгоритмов
|
||||
Создание CSV и графиков
|
||||
"""
|
||||
|
||||
import os
|
||||
import csv
|
||||
from datetime import datetime
|
||||
from maze_model import Maze
|
||||
from maze_builder import TextFileMazeBuilder
|
||||
from pathfinding_strategies import BFSStrategy, DFSStrategy, AStarStrategy
|
||||
from maze_solver import MazeSolver, SearchStats
|
||||
|
||||
|
||||
class ExperimentRunner:
|
||||
def __init__(self):
|
||||
self.all_results = []
|
||||
self.labirints = {
|
||||
'labirint1.txt': 'Маленький (10x10) с простым путём',
|
||||
'labirint2.txt': 'Средний (50x50) с тупиками',
|
||||
'labirint3.txt': 'Большой (100x100) запутанный',
|
||||
'labirint4.txt': 'Пустой (20x20) без стен',
|
||||
'labirint5.txt': 'Без выхода (20x20)'
|
||||
}
|
||||
self.strategies = [
|
||||
BFSStrategy(),
|
||||
DFSStrategy(),
|
||||
AStarStrategy()
|
||||
]
|
||||
|
||||
def run_all_experiments(self):
|
||||
"""Запускает эксперименты для всех лабиринтов и алгоритмов"""
|
||||
print("\n" + "="*70)
|
||||
print("ЗАПУСК ЭКСПЕРИМЕНТОВ")
|
||||
print("="*70)
|
||||
|
||||
builder = TextFileMazeBuilder()
|
||||
|
||||
for filename, description in self.labirints.items():
|
||||
if not os.path.exists(filename):
|
||||
print(f"\n⚠️ Файл {filename} не найден, пропускаем...")
|
||||
continue
|
||||
|
||||
print(f"\n📁 Лабиринт: {description}")
|
||||
print(f" Файл: {filename}")
|
||||
print("-" * 50)
|
||||
|
||||
try:
|
||||
maze = builder.build_from_file(filename)
|
||||
maze_name = filename.replace('.txt', '')
|
||||
|
||||
for strategy in self.strategies:
|
||||
print(f" Тестирование {strategy.get_name()}...", end=" ", flush=True)
|
||||
|
||||
solver = MazeSolver(maze, maze_name, strategy)
|
||||
path, stats = solver.solve_with_stats()
|
||||
|
||||
self.all_results.append({
|
||||
'лабиринт': description,
|
||||
'стратегия': stats.algorithm_name,
|
||||
'время_мс': stats.execution_time_ms,
|
||||
'посещено_клеток': stats.visited_cells,
|
||||
'длина_пути': stats.path_length,
|
||||
'путь_найден': 'Да' if stats.path_found else 'Нет',
|
||||
'размер': stats.maze_size
|
||||
})
|
||||
|
||||
print(f"готово! время={stats.execution_time_ms:.3f}мс, путь={stats.path_length}")
|
||||
|
||||
except Exception as e:
|
||||
print(f" ❌ Ошибка: {e}")
|
||||
|
||||
print("\n" + "="*70)
|
||||
print("ЭКСПЕРИМЕНТЫ ЗАВЕРШЕНЫ")
|
||||
print("="*70)
|
||||
|
||||
def save_to_csv(self, filename="experiment_results.csv"):
|
||||
"""Сохраняет результаты в CSV"""
|
||||
if not self.all_results:
|
||||
print("Нет результатов для сохранения!")
|
||||
return
|
||||
|
||||
with open(filename, 'w', newline='', encoding='utf-8-sig') as f:
|
||||
fieldnames = ['лабиринт', 'стратегия', 'время_мс', 'посещено_клеток', 'длина_пути', 'путь_найден', 'размер']
|
||||
writer = csv.DictWriter(f, fieldnames=fieldnames)
|
||||
writer.writeheader()
|
||||
writer.writerows(self.all_results)
|
||||
|
||||
print(f"\n✅ Результаты сохранены в {filename}")
|
||||
|
||||
# Показываем содержимое CSV
|
||||
print("\n" + "="*70)
|
||||
print("СОДЕРЖИМОЕ CSV ФАЙЛА:")
|
||||
print("="*70)
|
||||
with open(filename, 'r', encoding='utf-8-sig') as f:
|
||||
print(f.read())
|
||||
|
||||
def create_charts(self):
|
||||
"""Создаёт графики для каждого лабиринта"""
|
||||
try:
|
||||
import matplotlib.pyplot as plt
|
||||
import numpy as np
|
||||
|
||||
print("\n" + "="*70)
|
||||
print("ПОСТРОЕНИЕ ГРАФИКОВ")
|
||||
print("="*70)
|
||||
|
||||
# Группируем результаты по лабиринтам
|
||||
results_by_maze = {}
|
||||
for result in self.all_results:
|
||||
maze = result['лабиринт']
|
||||
if maze not in results_by_maze:
|
||||
results_by_maze[maze] = []
|
||||
results_by_maze[maze].append(result)
|
||||
|
||||
# Для каждого лабиринта создаём отдельный график
|
||||
for maze_name, maze_results in results_by_maze.items():
|
||||
fig, axes = plt.subplots(1, 3, figsize=(15, 5))
|
||||
fig.suptitle(f'Сравнение алгоритмов: {maze_name}', fontsize=14, fontweight='bold')
|
||||
|
||||
algorithms = [r['стратегия'] for r in maze_results]
|
||||
|
||||
# График 1: Время выполнения
|
||||
times = [r['время_мс'] for r in maze_results]
|
||||
bars1 = axes[0].bar(algorithms, times, color=['blue', 'green', 'red'])
|
||||
axes[0].set_ylabel('Время (мс)')
|
||||
axes[0].set_title('Время выполнения')
|
||||
axes[0].tick_params(axis='x', rotation=15)
|
||||
for bar, val in zip(bars1, times):
|
||||
axes[0].text(bar.get_x() + bar.get_width()/2, bar.get_height() + 0.5,
|
||||
f'{val:.3f}', ha='center', va='bottom', fontsize=9)
|
||||
|
||||
# График 2: Посещённые клетки
|
||||
visited = [r['посещено_клеток'] for r in maze_results]
|
||||
bars2 = axes[1].bar(algorithms, visited, color=['blue', 'green', 'red'])
|
||||
axes[1].set_ylabel('Количество клеток')
|
||||
axes[1].set_title('Посещённые клетки')
|
||||
axes[1].tick_params(axis='x', rotation=15)
|
||||
for bar, val in zip(bars2, visited):
|
||||
axes[1].text(bar.get_x() + bar.get_width()/2, bar.get_height() + 1,
|
||||
f'{val:.0f}', ha='center', va='bottom', fontsize=9)
|
||||
|
||||
# График 3: Длина пути
|
||||
lengths = [r['длина_пути'] for r in maze_results]
|
||||
bars3 = axes[2].bar(algorithms, lengths, color=['blue', 'green', 'red'])
|
||||
axes[2].set_ylabel('Шагов')
|
||||
axes[2].set_title('Длина найденного пути')
|
||||
axes[2].tick_params(axis='x', rotation=15)
|
||||
for bar, val in zip(bars3, lengths):
|
||||
axes[2].text(bar.get_x() + bar.get_width()/2, bar.get_height() + 0.5,
|
||||
f'{val:.0f}', ha='center', va='bottom', fontsize=9)
|
||||
|
||||
plt.tight_layout()
|
||||
|
||||
# Сохраняем график
|
||||
safe_name = maze_name.replace(' ', '_').replace('(', '').replace(')', '').replace('×', 'x')
|
||||
filename = f"chart_{safe_name}.png"
|
||||
plt.savefig(filename, dpi=150, bbox_inches='tight')
|
||||
print(f"✅ Сохранён: {filename}")
|
||||
plt.close()
|
||||
|
||||
# Общий сводный график
|
||||
self._create_summary_chart()
|
||||
|
||||
except ImportError:
|
||||
print("\n⚠️ Для построения графиков установите matplotlib:")
|
||||
print(" pip install matplotlib numpy")
|
||||
|
||||
def _create_summary_chart(self):
|
||||
"""Создаёт сводный график по всем лабиринтам"""
|
||||
import matplotlib.pyplot as plt
|
||||
import numpy as np
|
||||
|
||||
fig, axes = plt.subplots(2, 2, figsize=(14, 10))
|
||||
fig.suptitle('Сводное сравнение алгоритмов по всем лабиринтам', fontsize=14, fontweight='bold')
|
||||
|
||||
# Получаем уникальные лабиринты и алгоритмы
|
||||
mazes = list(set([r['лабиринт'] for r in self.all_results]))
|
||||
algorithms = ['BFS (Поиск в ширину)', 'DFS (Поиск в глубину)', 'A* (A-Star)']
|
||||
|
||||
# 1. Время по лабиринтам
|
||||
ax1 = axes[0, 0]
|
||||
x = np.arange(len(mazes))
|
||||
width = 0.25
|
||||
|
||||
for i, algo in enumerate(algorithms):
|
||||
times = []
|
||||
for maze in mazes:
|
||||
result = next((r for r in self.all_results if r['лабиринт'] == maze and r['стратегия'] == algo), None)
|
||||
times.append(result['время_мс'] if result else 0)
|
||||
bars = ax1.bar(x + (i - 1) * width, times, width, label=algo)
|
||||
for bar, val in zip(bars, times):
|
||||
if val > 0:
|
||||
ax1.text(bar.get_x() + bar.get_width()/2, bar.get_height() + 0.5,
|
||||
f'{val:.1f}', ha='center', va='bottom', fontsize=7)
|
||||
|
||||
ax1.set_xlabel('Лабиринт')
|
||||
ax1.set_ylabel('Время (мс)')
|
||||
ax1.set_title('Сравнение времени выполнения')
|
||||
ax1.set_xticks(x)
|
||||
ax1.set_xticklabels([m[:20] for m in mazes], rotation=45, ha='right')
|
||||
ax1.legend()
|
||||
|
||||
# 2. Посещённые клетки
|
||||
ax2 = axes[0, 1]
|
||||
for i, algo in enumerate(algorithms):
|
||||
visited = []
|
||||
for maze in mazes:
|
||||
result = next((r for r in self.all_results if r['лабиринт'] == maze and r['стратегия'] == algo), None)
|
||||
visited.append(result['посещено_клеток'] if result else 0)
|
||||
bars = ax2.bar(x + (i - 1) * width, visited, width, label=algo)
|
||||
for bar, val in zip(bars, visited):
|
||||
if val > 0:
|
||||
ax2.text(bar.get_x() + bar.get_width()/2, bar.get_height() + 5,
|
||||
f'{val:.0f}', ha='center', va='bottom', fontsize=7)
|
||||
|
||||
ax2.set_xlabel('Лабиринт')
|
||||
ax2.set_ylabel('Посещённые клетки')
|
||||
ax2.set_title('Сравнение посещённых клеток')
|
||||
ax2.set_xticks(x)
|
||||
ax2.set_xticklabels([m[:20] for m in mazes], rotation=45, ha='right')
|
||||
ax2.legend()
|
||||
|
||||
# 3. Длина пути
|
||||
ax3 = axes[1, 0]
|
||||
for i, algo in enumerate(algorithms):
|
||||
lengths = []
|
||||
for maze in mazes:
|
||||
result = next((r for r in self.all_results if r['лабиринт'] == maze and r['стратегия'] == algo), None)
|
||||
lengths.append(result['длина_пути'] if result else 0)
|
||||
bars = ax3.bar(x + (i - 1) * width, lengths, width, label=algo)
|
||||
for bar, val in zip(bars, lengths):
|
||||
if val > 0:
|
||||
ax3.text(bar.get_x() + bar.get_width()/2, bar.get_height() + 1,
|
||||
f'{val:.0f}', ha='center', va='bottom', fontsize=7)
|
||||
|
||||
ax3.set_xlabel('Лабиринт')
|
||||
ax3.set_ylabel('Длина пути (шагов)')
|
||||
ax3.set_title('Сравнение длины пути')
|
||||
ax3.set_xticks(x)
|
||||
ax3.set_xticklabels([m[:20] for m in mazes], rotation=45, ha='right')
|
||||
ax3.legend()
|
||||
|
||||
# 4. Таблица результатов
|
||||
ax4 = axes[1, 1]
|
||||
ax4.axis('tight')
|
||||
ax4.axis('off')
|
||||
|
||||
# Создаём таблицу
|
||||
table_data = []
|
||||
for maze in mazes:
|
||||
row = [maze[:25]]
|
||||
for algo in algorithms:
|
||||
result = next((r for r in self.all_results if r['лабиринт'] == maze and r['стратегия'] == algo), None)
|
||||
if result:
|
||||
row.append(f"{result['время_мс']:.1f}мс")
|
||||
else:
|
||||
row.append("-")
|
||||
table_data.append(row)
|
||||
|
||||
columns = ['Лабиринт', 'BFS', 'DFS', 'A*']
|
||||
table = ax4.table(cellText=table_data, colLabels=columns, cellLoc='center', loc='center')
|
||||
table.auto_set_font_size(False)
|
||||
table.set_fontsize(9)
|
||||
table.scale(1.2, 1.5)
|
||||
ax4.set_title('Сводная таблица (время в мс)', fontsize=10)
|
||||
|
||||
plt.tight_layout()
|
||||
plt.savefig('summary_chart.png', dpi=150, bbox_inches='tight')
|
||||
print("Сохранён: summary_chart.png")
|
||||
plt.close()
|
||||
|
||||
|
||||
def main():
|
||||
|
||||
runner = ExperimentRunner()
|
||||
runner.run_all_experiments()
|
||||
runner.save_to_csv("experiment_results.csv")
|
||||
runner.create_charts()
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
18
osipovamd/maze_project/experiment_results.csv
Normal file
18
osipovamd/maze_project/experiment_results.csv
Normal file
|
|
@ -0,0 +1,18 @@
|
|||
лабиринт,стратегия,время_мс,длина_пути,путь_найден,размер,дата_время
|
||||
labirint1,BFS (Поиск в ширину),0.211,14,Да,10x10,2026-05-27 22:26:56
|
||||
labirint2,BFS (Поиск в ширину),0.858,0,Нет,50x51,2026-05-27 22:27:10
|
||||
labirint3,BFS (Поиск в ширину),0.055,0,Нет,100x100,2026-05-27 22:27:26
|
||||
labirint4,BFS (Поиск в ширину),1.326,35,Да,20x20,2026-05-27 22:27:35
|
||||
labirint5,BFS (Поиск в ширину),0.373,36,Да,21x20,2026-05-27 22:27:44
|
||||
labirint1,DFS (Поиск в глубину),0.135,14,Да,10x10,2026-05-27 22:28:16
|
||||
labirint2,DFS (Поиск в глубину),0.797,0,Нет,50x51,2026-05-27 22:28:25
|
||||
labirint3,DFS (Поиск в глубину),0.047,0,Нет,100x100,2026-05-27 22:28:31
|
||||
labirint4,DFS (Поиск в глубину),0.88,171,Да,20x20,2026-05-27 22:28:36
|
||||
labirint5,DFS (Поиск в глубину),0.772,36,Да,21x20,2026-05-27 22:28:41
|
||||
labirint1,A* (A-Star),0.311,14,Да,10x10,2026-05-27 22:28:45
|
||||
labirint2,A* (A-Star),1.318,0,Нет,50x51,2026-05-27 22:28:50
|
||||
labirint3,A* (A-Star),0.055,0,Нет,100x100,2026-05-27 22:28:55
|
||||
labirint4,A* (A-Star),2.301,35,Да,20x20,2026-05-27 22:29:00
|
||||
labirint5,A* (A-Star),0.684,36,Да,21x20,2026-05-27 22:29:04
|
||||
labirint1,A* (A-Star),0.316,14,Да,10x10,2026-05-27 22:36:50
|
||||
labirint1,DFS (Поиск в глубину),0.133,14,Да,10x10,2026-05-27 22:41:42
|
||||
|
Gitea Version: 1.22.0 |
