Update Feature_extraction_algorithms/PSTAAP.py

Feature_extraction_algorithms/PSTAAP.py

@@ -15,21 +15,23 @@ def load_precomputed_fr_matrix(mat_file_path: str):
     try:
         mat_data = scipy.io.loadmat(mat_file_path)
         
-        # --- !!! 这里是唯一的修改点 !!! ---
-        # 将 'Fr_train' 修改为 'Fr'，以匹配您的 .mat 文件
-        matrix_key = 'Fr' # <--- 修改这里！
+        matrix_key = 'Fr'  # 修改后的变量名
         
         if matrix_key not in mat_data:
             raise KeyError(f"在 {mat_file_path} 中未找到变量名 '{matrix_key}'。 "
                            f"文件中可用的变量有: {list(mat_data.keys())}")
             
         _cached_fr_matrix = mat_data[matrix_key]
-        
         _expected_length_after_processing = _cached_fr_matrix.shape[1] + 2
 
         print(f"Fr 矩阵加载并缓存成功。形状: {_cached_fr_matrix.shape}")
         print(f"推断出的序列期望长度 (处理后): {_expected_length_after_processing}")
 
+    except Exception as e:
+        print(f"❌ 加载 Fr 矩阵失败: {e}")
+        raise
+
+
 
 # --- 模块级缓存 ---
 # 这个变量将会在内存中存储计算好的Fr矩阵，避免重复计算和文件IO。