Jelajahi Sumber

feat:修改公历转农历

zhaohaipeng 2 bulan lalu
induk
melakukan
d9279a0118
2 mengubah file dengan 8 tambahan dan 5 penghapusan
  1. 3 3
      script/dnn_model_warm_up.py
  2. 5 2
      udf/solar_to_lunar.py

+ 3 - 3
script/dnn_model_warm_up.py

@@ -6,7 +6,7 @@ from eas_prediction import TFRequest
 shape = 5
 if __name__ == '__main__':
     fg_config = {}
-    with open("/Users/zhao/Downloads/feature_list_20260417.json", "r") as f:
+    with open("/Users/zhao/Downloads/feature_list_20260424.json", "r") as f:
         fg_config = json.load(f)
 
     req = TFRequest('serving_default')
@@ -19,7 +19,7 @@ if __name__ == '__main__':
         else:
             req.add_feed(feature_name, [shape], TFRequest.DT_STRING, [b"-1024"] * shape)
 
-    req.add_fetch('probs_is_share')
+    req.add_fetch('probs_pLeave')
 
-    with open("/Users/zhao/Desktop/warm_up_20260417.bin", "wb") as fw:
+    with open("/Users/zhao/Desktop/warm_up_20260424.bin", "wb") as fw:
         fw.write(req.to_string())

+ 5 - 2
udf/solar_to_lunar.py

@@ -168,8 +168,11 @@ def _lunar_to_string(lunar_year, lunar_month, lunar_day, is_leap, fmt_str, outpu
         lunar_year = _year_to_cnt(lunar_year)
         lunar_month = LUNAR_MONTH_NAME[lunar_month]
         lunar_day = LUNAR_DAY_NAME[lunar_day]
-    elif "MM" in fmt_str:
-        lunar_month = str(lunar_month).zfill(2)
+    else:
+        if "MM" in fmt_str:
+            lunar_month = str(lunar_month).zfill(2)
+        if "dd" in fmt_str:
+            lunar_day = str(lunar_day).zfill(2)
 
     ly = str(lunar_year)
     lm = str(lunar_month) if not is_leap else "闰" + str(lunar_month)