Function plot_complete_overview

utils/utils_data.py:577–613

创建完整的总览图

(all_results, rpm_names, retain_rates, output_dir, logger)

def plot_complete_overview(all_results, rpm_names, retain_rates, output_dir, logger):
    """
    创建完整的总览图
    """
    fig, axes = plt.subplots(2, 3, figsize=(18, 12))
    axes = axes.flatten()

    for rpm_idx, rpm_name in enumerate(rpm_names):
        if rpm_idx >= len(axes):
            break

        ax = axes[rpm_idx]

        if rpm_name in all_results:
            for retain_rate in retain_rates:
                if retain_rate in all_results[rpm_name]:
                    results = all_results[rpm_name][retain_rate]
                    if results:
                        srs = [r[0] for r in results]
                        test_accs = [r[2] for r in results]
                        ax.plot(srs, test_accs, marker='o', label=f'{retain_rate:.1f}', linewidth=1.5)

        ax.set_title(f'RPM {rpm_name}')
        ax.set_xlabel('Spectral Radius')
        ax.set_ylabel('Test Accuracy')
        ax.grid(True, alpha=0.3)
        ax.legend(title='Retain Rate', fontsize=8)

    # 隐藏多余的子图
    for i in range(len(rpm_names), len(axes)):
        axes[i].set_visible(False)

    plt.tight_layout()
    overview_path = output_dir / "complete_overview.png"
    plt.savefig(overview_path, dpi=300, bbox_inches='tight')
    plt.close()
    logger.info(f"Complete overview plot saved: {overview_path}")


def plot_loss_accuracy(file_path='loss.txt'):