Function create_scatterplot

scripts_python/profiling_statistics.py:357–392

(
    x_data: np.ndarray, y_data: np.ndarray, title: str, xlabel: str, ylabel: str, output_path: Path
)

def create_scatterplot(
    x_data: np.ndarray, y_data: np.ndarray, title: str, xlabel: str, ylabel: str, output_path: Path
):
    if len(x_data) == 0 or len(y_data) == 0:
        return

    fig, ax = plt.subplots(figsize=(10, 6))

    ax.scatter(x_data, y_data, alpha=0.6, color="blue", edgecolors="black")

    X = x_data.reshape(-1, 1)
    model = LinearRegression()
    model.fit(X, y_data)
    y_pred = model.predict(X)

    slope = model.coef_[0]
    r2 = r2_score(y_data, y_pred)
    mse = mean_squared_error(y_data, y_pred)

    ax.plot(
        x_data,
        y_pred,
        color="lightblue",
        linewidth=2,
        label=f"Slope={slope:.4f}, R²={r2:.4f}, MSE={mse:.4f}",
    )

    ax.set_xlabel(xlabel)
    ax.set_ylabel(ylabel)
    ax.set_title(title)
    ax.legend()
    ax.grid(True, alpha=0.3)

    plt.tight_layout()
    plt.savefig(output_path, dpi=150, bbox_inches="tight")
    plt.close()


def generate_base_charts(conn: sqlite3.Connection, output_dir: Path, pbar: tqdm):