xgb boost done

data/analyis/filetool.py

@@ -11,8 +11,8 @@ def buildDocument(path,filename):
     return Document()
 
 def saveDocument(document,path,filename):
-    if str.rfind(filename,0,3)!='doc':
-        if str.rfind(filename,0,4) !='docx':
+    if filename[-3:] != 'doc':
+        if filename[-4:] != 'docx':
             raise ValueError('{} is not a word file'.format(filename))
     return document.save(os.path.join(path,filename))
 

data/graph/drawplot.py

@@ -101,7 +101,8 @@ def draw_barplot(df,x,y,hue='',title=''):
     pltz = PyplotZ()
     pltz.enable_chinese()
     fig = plt.figure()
-    ax = fig.add_subplot(1, 1, 1)
+    # ax = fig.add_subplot(1, 1, 1)
+    plt.cla()
 
     sns.set(style="whitegrid")
     fig = plt.figure(figsize=(6, 4))
@@ -116,7 +117,7 @@ def draw_barplot(df,x,y,hue='',title=''):
     pltz.title(title)
     pltz.legend()
     plt.grid()
-    plt.show()
+    # plt.show()
     return fig