yake test updates
@@ -9,19 +9,21 @@ class TestYake(unittest.TestCase):
df = nlu.load('yake',verbose=True).predict('What a wonderful day! Arnold schwanenegger is the Terminator and he wants to get to the American chopper', metadata=True)
print(df.columns)
print(df)
- print(df[['keywords', 'keywords_score']])
+ print(df['keywords_sentence_confidence'])
+ print(df[['keywords', 'keywords_score_confidence']])
df = nlu.load('yake',verbose=True).predict('What a wonderful day! Arnold schwanenegger is the Terminator and he wants to get to the American chopper', output_level='token')
print(df.columns)
- print(df[['keywords', 'keywords_score']])
-
+ print(df[['keywords', 'keywords_score_confidence']])
+ print(df['keywords_sentence_confidence'])
df = nlu.load('yake',verbose=True).predict('What a wonderful day! Arnold schwanenegger is the Terminator and he wants to get to the American chopper', output_level='chunk')
print(df.columns)
- print(df[['keywords', 'keywords_score']])
+ print(df[['keywords', 'keywords_score_confidence']])
+ print(df['keywords_sentence_confidence'])
df = nlu.load('yake',verbose=True).predict('What a wonderful day! Arnold schwanenegger is the Terminator and he wants to get to the American chopper', output_level='document')
print(df.columns)
- print(df[['keywords', 'keywords_score']])
+ print(df[['keywords', 'keywords_score_confidence']])
if __name__ == '__main__':
unittest.main()
