ensemble.py 文件源码

def get_qid_valid_answer_dict(ques_file, adict):
    """
    Returns a dictionary mapping question IDs to valid neuron indices.
    """
    print 'Multiple choice mode: making valid answer dictionary...'
    valid_answer_dict = {}
    with open(ques_file, 'r') as f:
        qdata = json.load(f)
        for q in qdata['questions']:
            valid_answer_dict[q['question_id']] = q['multiple_choices']
    for qid in valid_answer_dict:
        answers = valid_answer_dict[qid]
        valid_indices = []
        for answer in answers:
            if answer in adict:
                valid_indices.append(adict[answer])
        if len(valid_indices) == 0:
            print "we won't be able to answer qid", qid
        valid_answer_dict[qid] = valid_indices
    return valid_answer_dict