Gen TANG 2 년 전
부모
커밋
af872ca8ce
2개의 변경된 파일10개의 추가작업 그리고 10개의 파일을 삭제
  1. 6 6
      ch12_rl/utils.py
  2. 4 4
      ch13_others/viterbipy.py

+ 6 - 6
ch12_rl/utils.py

@@ -1,7 +1,7 @@
 # -*- coding: UTF-8 -*-
-"""
+'''
 此脚本用于定义游戏以及相应的可视化工具
-"""
+'''
 
 
 import matplotlib.pyplot as plt
@@ -41,7 +41,7 @@ class Lottery:
 
 def plot_values(v):
     # 为在Matplotlib中显示中文,设置特殊字体
-    plt.rcParams["font.sans-serif"] = ["SimHei"]
+    plt.rcParams['font.sans-serif'] = ['SimHei']
     # 正确显示负号
     plt.rcParams['axes.unicode_minus'] = False
     plt.rcParams.update({'font.size': 13})
@@ -50,14 +50,14 @@ def plot_values(v):
     v = pd.DataFrame(v)
     for k in v:
         v[k].plot(label=k, legend=True)
-    legend = plt.legend(shadow=True, loc="best", fontsize=20)
+    legend = plt.legend(shadow=True, loc='best', fontsize=20)
     plt.yticks(range(-10, 11, 4))
     return fig
 
 
 def plot_action_probs(v):
     # 为在Matplotlib中显示中文,设置特殊字体
-    plt.rcParams["font.sans-serif"] = ["SimHei"]
+    plt.rcParams['font.sans-serif'] = ['SimHei']
     # 正确显示负号
     plt.rcParams['axes.unicode_minus'] = False
     plt.rcParams.update({'font.size': 13})
@@ -67,5 +67,5 @@ def plot_action_probs(v):
     for k in v:
         # 在图中画出抽奖的概率
         v[k].apply(lambda x: x[1]).plot(label=k, legend=True)
-    legend = plt.legend(shadow=True, loc="best", fontsize=20)
+    legend = plt.legend(shadow=True, loc='best', fontsize=20)
     return fig

+ 4 - 4
ch13_others/viterbipy.py

@@ -1,7 +1,7 @@
 # -*- coding: UTF-8 -*-
-"""
+'''
 此脚本用于实现viterbi算法
-"""
+'''
 
 
 import numpy as np
@@ -9,7 +9,7 @@ from sklearn.utils.extmath import safe_sparse_dot
 
 
 def viterbi(obs, init_prob, trans_prob, emit_prob):
-    """
+    '''
     viterbi算法
 
     参数
@@ -31,7 +31,7 @@ def viterbi(obs, init_prob, trans_prob, emit_prob):
     score : {np.array},维度为(样本数,状态数),viterbi算法中间概率
 
     path : {np.array},维度为(样本数),最终结果表示每个样本的隐藏状态
-    """
+    '''
     sample_num, state_num = obs.shape[0], init_prob.shape[0]
     backp = np.empty((sample_num, state_num), dtype=np.intp)
     score = safe_sparse_dot(obs, emit_prob.T)