111

app/plot.py

@@ -1,3 +1,9 @@
+"""
+File: config.py
+Author: Elena Ryumina and Dmitry Ryumin
+Description: Plotting statistical information.
+License: MIT License
+"""
 import matplotlib.pyplot as plt
 import numpy as np 
 
@@ -20,4 +26,4 @@ def statistics_plot(frames, probs):
     ax.set_xlabel('Frames', fontsize=12)  # Добавляем подпись к оси X
     ax.set_ylabel('Probability', fontsize=12)  # Добавляем подпись к оси Y
     ax.grid(True)
-    return plt
+    return plt