自动创建matrix文件夹

app.py

@@ -54,8 +54,9 @@ def run():
     parser.add_argument("--run_itemcf_inference", action='store_true')
     
     # 协同过滤相关配置
-    parser.add_argument("--interst_score_path", type=str, default="./models/recall/itemCF/matrix/score.csv")
-    parser.add_argument("--similarity_matrix_path", type=str, default="./models/recall/itemCF/matrix/similarity.csv")
+    parser.add_argument("--matrix_path", type=str, default="./models/recall/itemCF/matrix")
+    # parser.add_argument("--interst_score_path", type=str, default="./models/recall/itemCF/matrix/score.csv")
+    # parser.add_argument("--similarity_matrix_path", type=str, default="./models/recall/itemCF/matrix/similarity.csv")
     parser.add_argument("--n", type=int, default=100)
     parser.add_argument("--k", type=int, default=10)
     parser.add_argument("--top_n", type=int, default=200, help='default n * k')
@@ -66,6 +67,13 @@ def run():
     
     args = parser.parse_args()
     
+    # 初始化文件保存相关配置
+    if not os.path.exists(args.matrix_path):
+        os.makedirs(args.matrix_path)
+    args.interst_score_path = os.path.join(args.matrix_path, "score.csv")
+    args.similarity_matrix_path = os.path.join(args.matrix_path, "similarity.csv")
+    
+    
     if args.run_all:
         order_data = load_order_data_from_mysql()
         run_hot_recall(order_data)