labels_tools.py 文件源码

def getLabelDetails(experiment, instance_id):
    if experiment.labels_type == 'partial_labels':
        query = experiment.session.query(LabelsAlchemy)
        query = query.filter(LabelsAlchemy.instance_id == int(instance_id))
        query = query.filter(LabelsAlchemy.labels_id == int(experiment.labels_id))
        try:
            res = query.one()
            return res.label, res.family, res.method, res.annotation
        except NoResultFound:
            return None
    if experiment.labels_type == 'true_labels':
        query = experiment.session.query(TrueLabelsAlchemy)
        query = query.filter(TrueLabelsAlchemy.instance_id == int(instance_id))
        try:
            res = query.one()
            return res.label, res.family, 'true_labels', True
        except NoResultFound:
            return None
    return None