iOS NSLinguisticTagger всегда возвращает ноль при получении основы леммы
Мне нужно сопоставить ключевое слово (в его лексической форме) с текстом и вернуть, совпадает ли ключевое слово (как поисковая система)
например
text: I want some apples
keyword: apple
result: matched
keyword: banana
result: not matched
Это мой метод, я всегда получаю нулевое значение для тега
- (BOOL)matchKeywordWithText:(NSString *)text keyword:(NSString *)keyword {
__block NSMutableArray *words = [NSMutableArray array];
NSLinguisticTaggerOptions options = NSLinguisticTaggerOmitWhitespace | NSLinguisticTaggerOmitPunctuation | NSLinguisticTaggerJoinNames;
NSLinguisticTagger *tagger = [[NSLinguisticTagger alloc] initWithTagSchemes: [NSLinguisticTagger availableTagSchemesForLanguage:@"en"] options:options];
tagger.string = text;
[tagger enumerateTagsInRange:NSMakeRange(0, [text length]) scheme:NSLinguisticTagSchemeLemma options:options usingBlock:^(NSString *tag, NSRange tokenRange, NSRange sentenceRange, BOOL *stop) {
NSString *token = [text substringWithRange:tokenRange];
//TODO: TAG IS ALWAYS NULL HERE
NSLog(@"words are %@: %@", token, tag);
if (tag.length > 0) {
[words addObject:tag];
}
}];
for (NSString *word in words) {
if ([word isEqualToString:keyword]) {
return YES;
}
}
return NO;
}
1 ответ
Решение
Работает следующий код:
NSLinguisticTagger *tagger = [[NSLinguisticTagger alloc]
initWithTagSchemes:[NSArray arrayWithObjects:NSLinguisticTagSchemeNameTypeOrLexicalClass, NSLinguisticTagSchemeLemma, nil]
options:(NSLinguisticTaggerOmitWhitespace | NSLinguisticTaggerOmitPunctuation)];
NSString *text = @"i'm apples";
[tagger setString:text];
[tagger enumerateTagsInRange:NSMakeRange(0, [text length])
scheme:NSLinguisticTagSchemeNameTypeOrLexicalClass
options:(NSLinguisticTaggerOmitWhitespace | NSLinguisticTaggerOmitPunctuation)
usingBlock:^(NSString *tag, NSRange tokenRange, NSRange sentenceRange, BOOL *stop) {
NSString *token = [text substringWithRange:tokenRange];
NSString *lemma = [tagger tagAtIndex:tokenRange.location scheme:NSLinguisticTagSchemeLemma tokenRange:NULL sentenceRange:NULL];
if (lemma == nil) {
lemma = token;
}
NSLog(@"%@, %@", token, lemma);
}];