33 lines
883 B
Python
33 lines
883 B
Python
|
import json
|
|||
|
|
|
|||
|
|
from gnnrec.config import DATA_DIR
|
|||
|
|
|
|||
|
|
|
|||
|
|
class Context:
|
|||
|
|
|
|||
|
|
def __init__(self, recall_ctx, author_rank):
|
|||
|
|
"""学者排名模块上下文对象
|
|||
|
|
|
|||
|
|
:param recall_ctx: gnnrec.kgrec.recall.Context
|
|||
|
|
:param author_rank: {field_id: [author_id]} 领域学者排名
|
|||
|
|
"""
|
|||
|
|
self.recall_ctx = recall_ctx
|
|||
|
|
# 之后需要：author_embeds
|
|||
|
|
self.author_rank = author_rank
|
|||
|
|
|
|||
|
|
|
|||
|
|
def get_context(recall_ctx):
|
|||
|
|
with open(DATA_DIR / 'rank/author_rank_train.json') as f:
|
|||
|
|
author_rank = json.load(f)
|
|||
|
|
return Context(recall_ctx, author_rank)
|
|||
|
|
|
|||
|
|
|
|||
|
|
def rank(ctx, query):
|
|||
|
|
"""根据输入的查询词在oag-cs数据集计算学者排名
|
|||
|
|
|
|||
|
|
:param ctx: Context 上下文对象
|
|||
|
|
:param query: str 查询词
|
|||
|
|
:return: List[float], List[int] 学者得分和id，按得分降序排序
|
|||
|
|
"""
|
|||
|
|
return [], ctx.author_rank.get(query, [])
|