-
Notifications
You must be signed in to change notification settings - Fork 37
Expand file tree
/
Copy pathcategories.py
More file actions
48 lines (35 loc) · 1.67 KB
/
categories.py
File metadata and controls
48 lines (35 loc) · 1.67 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
# -*- coding: utf-8 -*-
"""
Example code to call Analytics API to get the category of a document (at a given URL).
"""
import argparse
import json
import os
from rosette.api import API, DocumentParameters, RosetteException
def run(key, alt_url='https://analytics.babelstreet.com/rest/v1/'):
""" Run the example """
categories_text_data = "If you are a fan of the British television series Downton Abbey and you are planning to be in New York anytime before April 2nd, there is a perfect stop for you while in town."
# Create an API instance
api = API(user_key=key, service_url=alt_url)
# Set selected API options
# For more information on the functionality of these
# and other available options, see Analytics Features & Functions
# https://developer.babelstreet.com/features-and-functions#categorization
# api.set_option('singleLabel', 'true')
# api.set_option('scoreThreshold',- 0.20)
params = DocumentParameters()
params["content"] = categories_text_data
try:
return api.categories(params)
except RosetteException as exception:
print(exception)
PARSER = argparse.ArgumentParser(formatter_class=argparse.ArgumentDefaultsHelpFormatter,
description='Calls the ' +
os.path.splitext(os.path.basename(__file__))[0] + ' endpoint')
PARSER.add_argument('-k', '--key', help='Analytics API Key', required=True)
PARSER.add_argument('-u', '--url', help="Alternative API URL",
default='https://analytics.babelstreet.com/rest/v1/')
if __name__ == '__main__':
ARGS = PARSER.parse_args()
RESULT = run(ARGS.key, ARGS.url)
print(RESULT)