Извлечение определенного типа слова из абзаца с использованием обработки естественного языка

Question

Я разрабатываю модель, которая извлекает слова определенного типа.рассмотрим мой набор данных следующим образом:

  1. I want to book movie tickets for 2 peoples.
  2. I need to book movie tickets for 2 seats.
  3. I required two seat for movie

из трех приведенных выше утверждений, я хочу извлечь число 2, оно может быть целым или строковым или относится к местам или людям.

Я пробовал распознавание именованных объектов, но не получил требуемого вывода.и для другого вывода я использовал сентиментальный анализ.но проблема состоит в том, чтобы извлечь количество людей.

Я ожидаю количество людей из абзаца, которое может быть целым или строковым.Спасибо за помощь.

Answer 1

import spacy
nlp = spacy.load('en')   
doc = nlp("I want to book movie tickets for 2 peoples. I need to book movie tickets
           for 2 seats. I required two seat for movie.")
for i in doc.ents:
if i.label_ == 'CARDINAL':
    print(i.text, i.label_)

Выход: -

2 CARDINAL
2 CARDINAL
two CARDINAL