add simple confusion matrix

draw_confusion.py

@@ -0,0 +1,25 @@
+# https://www.cnblogs.com/yexionglin/p/11432180.html
+
+import seaborn as sns
+from sklearn.metrics import confusion_matrix
+import matplotlib.pyplot as plt
+
+
+def draw(y_true, y_pred):
+	sns.set()
+	f, ax=plt.subplots()
+	# y_true = [0,0,1,2,1,2,0,2,2,0,1,1]
+	# y_pred = [1,0,1,2,1,0,0,2,2,0,1,1]
+	cm = confusion_matrix(y_true, y_pred, labels=[-1] * 13)
+	print(cm) #打印出来看看
+	sns.heatmap(cm, annot=True, ax=ax) #画热力图
+
+def draw2(y_true, y_pred):
+	min_len = min( len(y_true), len(y_pred) )
+	for i in range(min_len):
+		for j in range(i):
+			print('\t', end='')
+		print(str(y_pred[i]))
+
+draw2( [1.0] * 13, [0.5] * 13 )
+

test2.py → test_dreamtalk.py

@@ -1,14 +1,43 @@
 import requests
+import sys
+from draw_confusion import draw, draw2
 
 API_URL = "https://api-inference.huggingface.co/models/MIT/ast-finetuned-audioset-10-10-0.4593"
 headers = {"Authorization": "Bearer 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"}
 
-filename = '1.flac'
+# filename = '1.flac'
 def query(filename):
     with open(filename, "rb") as f:
         data = f.read()
     response = requests.post(API_URL, headers=headers, data=data)
     return response.json()
 
-res = query(filename)
-print(str(res))
+# res = query(filename)
+# print(str(res))
+
+# 处理命令行
+if __name__ == "__main__":
+    # 获取命令行参数
+    if len(sys.argv) < 2:
+        print("用法：python x.py <文件或通配符>")
+        sys.exit(1)
+
+    y_len = 13 #len(sys.argv[1:]:)
+    y_true = [0] * y_len
+    y_pred = [0] * y_len
+    y_idx = 0
+    for input_file in sys.argv[1:]:
+        res = query(input_file)
+        print('%s:' % str(input_file))
+        print('%s' % str(res[:3]))
+        
+        first_label = str(res[0]['label'])
+        first_score = res[0]['score']
+        print(str(first_label))
+        print(str(first_score))
+
+        y_true[y_idx] = 1.0
+        y_pred[y_idx] = round(first_score, 1)
+        y_idx = y_idx + 1
+    
+    draw2(y_true, y_pred)