xlin 0.1.28__py2.py3-none-any.whl → 0.1.29__py2.py3-none-any.whl

xlin/statistic.py

@@ -111,11 +111,53 @@ Kurtosis: {float((data - mean).mean()**4 / std**4):.4f}\
     plt.show()
 
 
-def draw_pie(numbers: List[int], title="Pie Chart of Numbers"):
+
+def draw_preds_labels(preds: list[str], labels: list[str]):
+    from collections import Counter
     import matplotlib.pyplot as plt
 
-    plt.pie(numbers, labels=[str(i) for i in range(len(numbers))], autopct='%1.1f%%')
-    plt.title(title)
+    out_of_class = "out_of_class"
+    valid_values = list(set(labels)) + [out_of_class]
+    valid_preds = []
+    for pred in preds:
+        if pred not in valid_values:
+            valid_preds.append(out_of_class)
+        else:
+            valid_preds.append(pred)
+
+    counter = Counter(valid_preds)
+    pred_labels = list(counter.keys())
+    pred_values = list(counter.values())
+
+    # 绘制柱状图 pred
+    plt.figure(figsize=(12, 12))
+    plt.subplot(2, 2, 1)
+    plt.bar(pred_labels, pred_values)
+    plt.xlabel("class")
+    plt.ylabel("count")
+    plt.title("pred class distribution")
+
+    # 绘制饼图 pred
+    plt.subplot(2, 2, 2)
+    plt.pie(pred_values, labels=pred_labels, autopct="%1.1f%%")
+    plt.title("pred class distribution")
+
+    # 绘制柱状图 label
+    counter = Counter(labels)
+    label_labels = list(counter.keys())
+    label_values = list(counter.values())
+    plt.subplot(2, 2, 3)
+    plt.bar(label_labels, label_values)
+    plt.xlabel("class")
+    plt.ylabel("count")
+    plt.title("label class distribution")
+    # 绘制饼图 label
+    plt.subplot(2, 2, 4)
+    plt.pie(label_values, labels=label_labels, autopct="%1.1f%%")
+    plt.title("label class distribution")
+    plt.suptitle("Pred and Label Class Distribution")
+
+    plt.tight_layout()
     plt.show()
 
 
@@ -256,6 +298,7 @@ def print_classification_report(predictions: List[str], labels: List[str]):
     print("=== 分类报告 ===")
     print(report["class_report"])
     print()
+    return report
 
 
 if __name__ == "__main__":

{xlin-0.1.28.dist-info → xlin-0.1.29.dist-info}/METADATA

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

{xlin-0.1.28.dist-info → xlin-0.1.29.dist-info}/RECORD

@@ -4,12 +4,12 @@ xlin/jsonl.py,sha256=IDRydHh2x-8iAGCxt9ScK2wfNLNA40PxNxR5hhr4v6k,7903
 xlin/metric.py,sha256=N7wJ35y-C-IaBr1I1CJ_37lTG7gA69zmn9Xg6xSwKoI,1690
 xlin/multiprocess_mapping.py,sha256=q4EVU8JPLcRAbNf9NUEzmn8rDLDfIQ3jaW0yxVKPECk,16669
 xlin/read_as_dataframe.py,sha256=MqY57L7Wp9UoWTRlZLSBKQNaZa-dKw51-ufrKvHKf8s,9041
-xlin/statistic.py,sha256=2DCUgzf7xkMFH4Pk9v82bFDNeSxCTjwPh9Y4IPJBHCE,9300
+xlin/statistic.py,sha256=VS8tL0rwhs3tBR0hC4IU-7FVIlHuxImMrqJzut8JKNI,10525
 xlin/timing.py,sha256=XMT8dMcMolOMohDvAZOIM_BAiPMREhGQKnO1kc5s6PU,998
 xlin/util.py,sha256=TTWJaqF5D_r-gAZ_fj0kyHomvCagjwHXQZ2OPSgwd54,10976
 xlin/xls2xlsx.py,sha256=uSmXcDvIhi5Sq0LGidMXy0wErNBXdjaoa6EftYVjTXs,947
 xlin/yaml.py,sha256=kICi7G3Td5q2MaSXXt85qNTWoHMgjzt7pvn7r3C4dME,183
-xlin-0.1.28.dist-info/LICENSE,sha256=60ys6rRtc1dZOP8UjSUr9fAqhZudT3WpKe5WbMCralM,1066
-xlin-0.1.28.dist-info/METADATA,sha256=Ft8sYxlqCYTiuej-KTNXrKbFGhahW0LOSdng3XWH7Z0,1098
-xlin-0.1.28.dist-info/WHEEL,sha256=IrRNNNJ-uuL1ggO5qMvT1GGhQVdQU54d6ZpYqEZfEWo,92
-xlin-0.1.28.dist-info/RECORD,,
+xlin-0.1.29.dist-info/LICENSE,sha256=60ys6rRtc1dZOP8UjSUr9fAqhZudT3WpKe5WbMCralM,1066
+xlin-0.1.29.dist-info/METADATA,sha256=oGtT05fTogUwHLtNFuLkmV8oxd_wzLIJfC69f6Q4JYg,1098
+xlin-0.1.29.dist-info/WHEEL,sha256=IrRNNNJ-uuL1ggO5qMvT1GGhQVdQU54d6ZpYqEZfEWo,92
+xlin-0.1.29.dist-info/RECORD,,