Method save_results_to_csv

utils/quantize_embeddings.py:329–368

将benchmark结果保存到CSV文件

(self)

        self.print_summary(total_duration)
    
    def save_results_to_csv(self):
        """将benchmark结果保存到CSV文件"""
        if not self.results:
            print("⚠️  No results to save")
            return
        
        # Use user-specified CSV path, otherwise use default path
        if self.csv_output:
            csv_file = self.csv_output
            # Ensure parent directory exists
            csv_file.parent.mkdir(parents=True, exist_ok=True)
        else:
            csv_file = self.stats_dir / f"embedding_benchmark.csv"
        
        print(f"\n💾 Saving results to: {csv_file}")
        
        try:
            with open(csv_file, 'w', newline='') as f:
                fieldnames = ['embedding_type', 'threads_1', 'threads_2', 'threads_4', 'threads_8']
                writer = csv.DictWriter(f, fieldnames=fieldnames)
                
                writer.writeheader()
                for result in self.results:
                    writer.writerow(result)
            
            print(f"✅ Results saved successfully")
            
            # Also print table
            print(f"\n📊 Benchmark Results:")
            print(f"{'Type':<15} {'1 thread':<18} {'2 threads':<18} {'4 threads':<18} {'8 threads':<18}")
            print("-" * 87)
            for result in self.results:
                t1 = result['threads_1'] if result['threads_1'] else "N/A"
                t2 = result['threads_2'] if result['threads_2'] else "N/A"
                t4 = result['threads_4'] if result['threads_4'] else "N/A"
                t8 = result['threads_8'] if result['threads_8'] else "N/A"
                print(f"{result['embedding_type']:<15} {t1:<18} {t2:<18} {t4:<18} {t8:<18}")
                
        except Exception as e:
            print(f"❌ Failed to save results: {e}")
        
    def print_summary(self, total_duration):
        """Print quantization summary"""