def get_article_representation(self, article_title):
''' Returns a list of ids of the categories of this article
@return: a list of numbers '''
category_titles = wikipedia_api_util.query_categories_of_res(article_title)
print "categories of "+str(article_title)+" is "+str(category_titles)
category_ids = [wikipedia_api_util.query_page_id(cat_title) for cat_title in category_titles]
return category_ids
def get_article_representation(self, article_title):
''' Returns a list containing the id of this article
@return: a list of numbers '''
article_model = [wikipedia_api_util.query_page_id(article_title)]
return article_model
