cnn_tensorflow.py 文件源码

def save_dataset(dataset, output_path):
    """ Save the whole dataset as pickle file
    :param dataset: pandas DataFrame
    :param output_path: path adn file name of output
    """
    logging.info('Saving dataset to pickle file :' + output_path)

    os.makedirs(os.path.dirname(output_path), exist_ok=True)
    pandas.to_pickle(dataset, output_path)