sklearn.textdata.datasets.load_files

import numpy as np
import pandas as pd
import matplotlib.pyplot as plt

 

from sklearn.datasets import load_files

# 파일 읽기
imdb_tarin=load_files('data/aclImdb/train')
imdb_test=load_files('data/aclImdb/test')


# numpy로 저장(이후에 빨리 읽기 위해)
np.save('imdb.npy',[imdb_tarin,imdb_test])

 

# numpy 파일 읽기

imdb_tarin,imdb_test=np.load('imdb.npy')