util.py 文件源码

def get_pub_dic_csv(dataset):
    filename = "data/" + dataset + "-text.csv"
    f = open(filename)
    f.readline()
    csv_reader = csv.reader(f)

    # Create dic of : id -> text features
    pub_dic = {}

    for row in csv_reader:
        if dataset.startswith("RCT"):
            (abstract_id, abstract, title) = tuple(row)[0:3]
        else:
            (abstract_id, title, publisher, abstract) = tuple(row)[0:4]

        abstract_id = int(abstract_id)
        text = title + abstract

        pub_dic[abstract_id] = text

    return pub_dic