def __init__(self, chatbot, **kwargs):
super().__init__(chatbot, **kwargs)
try:
from nltk import NaiveBayesClassifier
except ImportError:
message = (
'Unable to import "nltk".\n'
'Please install "nltk" before using the TimeLogicAdapter:\n'
'pip3 install nltk')
raise OptionalDependencyImportError(message)
self.positive = kwargs.get('positive', [
'what time is it', 'hey what time is it', 'do you have the time',
'do you know the time', 'do you know what time it is',
'what is the time'
])
self.negative = kwargs.get('negative', [
'it is time to go to sleep', 'what is your favorite color',
'i had a great time', 'thyme is my favorite herb',
'do you have time to look at my essay',
'how do you have the time to do all this'
'what is it'
])
labeled_data = ([(name, 0) for name in self.negative] +
[(name, 1) for name in self.positive])
train_set = [(self.time_question_features(text), n)
for (text, n) in labeled_data]
self.classifier = NaiveBayesClassifier.train(train_set)
def __init__(self, language):
super().__init__(language)
try:
import spacy
except ImportError:
message = (
'Unable to import "spacy".\n'
'Please install "spacy" before using the SpacySimilarity comparator:\n'
'pip3 install "spacy>=2.1,<2.2"')
raise OptionalDependencyImportError(message)
self.nlp = spacy.load('en_core_web_lg')
def __init__(self, chatbot, **kwargs):
super().__init__(chatbot, **kwargs)
try:
from pint import UnitRegistry
except ImportError:
message = (
'Unable to import "pint".\n'
'Please install "pint" before using the UnitConversion logic adapter:\n'
'pip3 install pint'
)
raise OptionalDependencyImportError(message)
self.unit_registry = UnitRegistry()
self.language = kwargs.get('language', languages.ENG)
self.cache = {}
self.patterns = [
(
re.compile(r'''
(([Hh]ow\s+many)\s+
(?P<target>\S+)\s+ # meter, celsius, hours
((are)*\s*in)\s+
(?P<number>([+-]?\d+(?:\.\d+)?)|(a|an)|(%s[-\s]?)+)\s+
(?P<from>\S+)\s*) # meter, celsius, hours
''' % (parsing.numbers),
(re.VERBOSE | re.IGNORECASE)
),
lambda m: self.handle_matches(m)
),
(
re.compile(r'''
((?P<number>([+-]?\d+(?:\.\d+)?)|(%s[-\s]?)+)\s+
(?P<from>\S+)\s+ # meter, celsius, hours
(to)\s+
(?P<target>\S+)\s*) # meter, celsius, hours
''' % (parsing.numbers),
(re.VERBOSE | re.IGNORECASE)
),
lambda m: self.handle_matches(m)
),
(
re.compile(r'''
((?P<number>([+-]?\d+(?:\.\d+)?)|(a|an)|(%s[-\s]?)+)\s+
(?P<from>\S+)\s+ # meter, celsius, hours
(is|are)\s+
(how\s+many)*\s+
(?P<target>\S+)\s*) # meter, celsius, hours
''' % (parsing.numbers),
(re.VERBOSE | re.IGNORECASE)
),
lambda m: self.handle_matches(m)
)
]
def read_corpus(file_name):
"""
Read and return the data from a corpus json file.
"""
try:
import yaml
except ImportError:
message = (
'Unable to import "yaml".\n'
'Please install "pyyaml" to enable chatterbot corpus functionality:\n'
'pip3 install pyyaml')
raise OptionalDependencyImportError(message)
with io.open(file_name, encoding='utf-8') as data_file:
return yaml.safe_load(data_file)