DataIO.py 文件源码

def LoadFromPklFile(InputDir):

        with open('%s/train.pkl' % InputDir, 'rb') as i_file:
            TrainData = pickle.load(i_file)
        i_file.close()

        with open('%s/test.pkl' % InputDir, 'rb') as i_file:
            TestData = pickle.load(i_file)
        i_file.close()

        return TrainData,TestData

    ## class method, load data with text format