随机选取

barrage/analysisEmotion.py

@@ -15,6 +15,7 @@ ps.柱形图的可能性是指情感识别为真的平均可能性，不是弹
 
 from paddlenlp import Taskflow
 import matplotlib.pyplot as plt
+import random
 
 # 加载弹幕字符文本
 def loadText(sep, filePath):
@@ -71,10 +72,11 @@ def main():
     for i in range(1000):
         if i%100 ==0:
             print(f'当前正在处理第{i}条弹幕')
-        if schema not in ie(t_list[i])[0]:
+        num = random.randint(0, 20000)
+        if schema not in ie(t_list[num])[0]:
             continue
-        emo = ie(t_list[i])[0][schema][0]['text']
-        pro = ie(t_list[i])[0][schema][0]['probability']
+        emo = ie(t_list[num])[0][schema][0]['text']
+        pro = ie(t_list[num])[0][schema][0]['probability']
         emoChange(emo, pro, count, probability)
 
     createBar(count, probability, savePath)