mirror of
https://github.com/wassname/IndicoIo-python.git
synced 2026-06-27 16:10:34 +08:00
Madison May
26 lines
872 B
Python
26 lines
872 B
Python
from indicoio.utils import api_handler
|
|
|
|
def classification(api_root, text):
|
|
"""
|
|
Given input text, returns a probability distribution over 100 document categories
|
|
|
|
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.classification(text)
|
|
>>> category = possible.keys()[np.argmax(possible.values())]
|
|
>>> probability = np.max(possible.values())
|
|
>>> "Predicted category '%s' with probability %.4f"%(category,probability)
|
|
u'Predicted 'Weather' with probability 0.8548'
|
|
|
|
:param text: The text to be analyzed.
|
|
:type text: str or unicode
|
|
:rtype: Dictionary of class probability pairs
|
|
"""
|
|
|
|
return api_handler(text, api_root + "documentclassification")
|