mirror of
https://github.com/wassname/IndicoIo-python.git
synced 2026-06-28 16:20:42 +08:00
Madison May
35 lines
1.1 KiB
Python
35 lines
1.1 KiB
Python
import requests
|
|
import json
|
|
|
|
from indicoio import JSON_HEADERS
|
|
|
|
def language(api_root, text):
|
|
"""
|
|
Given input text, returns a probability distribution over 33 possible
|
|
languages of what language the text was written in.
|
|
|
|
Example usage:
|
|
|
|
.. code-block:: python
|
|
|
|
>>> import indicoio
|
|
>>> import numpy as np
|
|
>>> text = 'Monday: Delightful with mostly sunny skies. Highs in the low 70s.'
|
|
>>> possible = indicoio.language(text)
|
|
>>> language = possible.keys()[np.argmax(possible.values())]
|
|
>>> probability = np.max(possible.values())
|
|
>>> 'Predicted %s with probability %.4f'%(language,probability)
|
|
u'Predicted English with probability 0.8548'
|
|
|
|
:param text: The text to be analyzed.
|
|
:type text: str or unicode
|
|
:rtype: Dictionary of language probability pairs
|
|
"""
|
|
|
|
data_dict = json.dumps({'text': text})
|
|
response = requests.post(api_root + "language", data=data_dict, headers=JSON_HEADERS)
|
|
response_dict = response.json()
|
|
if len(response_dict) < 2:
|
|
raise ValueError(response_dict.values()[0])
|
|
else:
|
|
return response_dict |