Вступ до побудови конвеєра обробки тексту

Глибоке навчання для тексту з PyTorch

Shubham Jain

Data Scientist

Підсумок: попередня обробка

Конвеєр попередньої обробки

Попередня обробка:
- Токенізація
- Видалення стоп-слів
- Стемінг
- Видалення рідкісних слів

Конвеєр обробки тексту

Конвеєр кодування

Кодування:
- One-hot encoding
- Bag-of-words
- TF-IDF
Вбудовування

Конвеєр обробки тексту

Конвеєр обробки даних

Dataset як контейнер для обробленого й закодованого тексту
DataLoader: батчинг, перетасовка та мультипроцесинг

Підсумок: реалізація Dataset і DataLoader

# Import libraries
from torch.utils.data import Dataset, DataLoader

# Create a class
class TextDataset(Dataset):

    def __init__(self, text):
        self.text = text

    def __len__(self):
        return len(self.text)

    def __getitem__(self, idx):
        return self.text[idx]

Підсумок: інтеграція Dataset і DataLoader

dataset = TextDataset(encoded_text)

dataloader = DataLoader(dataset, batch_size=2, shuffle=True)

Використання допоміжних функцій

def preprocess_sentences(sentences):
    processed_sentences = []
    for sentence in sentences:
        sentence = sentence.lower()
        tokens = tokenizer(sentence)
        tokens = [token for token in tokens 
                  if token not in stop_words]
        tokens = [stemmer.stem(token) 
                  for token in tokens]
        freq_dist = FreqDist(tokens)
        threshold = 2
        tokens = [token for token in tokens if 
        freq_dist[token] > threshold]
        processed_sentences.append(
                   ' '.join(tokens))
    return processed_sentences

def encode_sentences(sentences):
    vectorizer = CountVectorizer()
    X = vectorizer.fit_transform(sentences)
    encoded_sentences = X.toarray()
    return encoded_sentences, vectorizer

def extract_sentences(data):
    sentences = re.findall(r'[A-Z][^.!?]*[.!?]', 
                           data)
    return sentences

Побудова конвеєра обробки тексту

def text_processing_pipeline(text):

    tokens = preprocess_sentences(text)

    encoded_sentences, vectorizer = encode_sentences(tokens)

    dataset = TextDataset(encoded_sentences)

    dataloader = DataLoader(dataset, batch_size=2, shuffle=True)

    return dataloader, vectorizer

Застосування конвеєра обробки тексту

text_data = "This is the first text data. And here is another one."

sentences = extract_sentences(text_data)
dataloaders, vectorizer = [text_processing_pipeline(text) for text in sentences]

print(next(iter(dataloader))[0, :10])

[[1, 1, 1, 1, 1], [0, 0, 0, 1, 1]]

Конвеєр обробки тексту: готово!

Повний конвеєр

Давайте потренуємось!

Глибоке навчання для тексту з PyTorch