xlin 0.1.29__tar.gz → 0.1.30__tar.gz

{xlin-0.1.29 → xlin-0.1.30}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: xlin
-Version: 0.1.29
+Version: 0.1.30
 Summary: toolbox for LinXueyuan
 License: MIT
 Author: LinXueyuanStdio

{xlin-0.1.29 → xlin-0.1.30}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "xlin"
-version = "0.1.29"
+version = "0.1.30"
 description = "toolbox for LinXueyuan"
 authors = ["LinXueyuanStdio <23211526+LinXueyuanStdio@users.noreply.github.com>"]
 license = "MIT"

{xlin-0.1.29 → xlin-0.1.30}/xlin/statistic.py

@@ -266,6 +266,20 @@ def generate_classification_report(predictions: List[str], labels: List[str]) ->
     return report
 
 
+def convert_to_jsonable_report(report_row):
+    new_report_json = {}
+    for key, value in report_row.items():
+        if isinstance(value, dict):
+            new_report_json[key] = convert_to_jsonable_report(value)
+        elif isinstance(value, list):
+            new_report_json[key] = [convert_to_jsonable_report(item) if isinstance(item, dict) else item for item in value]
+        elif isinstance(value, pd.DataFrame):
+            new_report_json[key] = value.fillna(-1).to_dict(orient="records")
+        else:
+            new_report_json[key] = value
+    return new_report_json
+
+
 def print_classification_report(predictions: List[str], labels: List[str]):
     report = generate_classification_report(predictions, labels)
     """