执行model.encode出错

[xyjsjruiliu]

代码：

coding: utf-8

import traceback import faulthandler faulthandler.enable()

from tqdm import *

import pandas as pd from bert4vec import Bert4Vec

model = Bert4Vec(mode='simbert-base', model_name_or_path="/Users/ruiliu/Downloads/simbert-base-chinese")

model = Bert4Vec(mode='roformer-sim-base', model_name_or_path="/Users/ruiliu/Downloads/roformer-sim-base-chinese")

def load_excel(excel_path, sheet_name=0): data_list = [] data = pd.read_csv(excel_path, keep_default_na=False) for i in data.index.values: data_list.append(data.loc[i].to_dict()) return data_list

if name == "main": try: data_list = load_excel("data/交互平台问答数据-ALL-CODE.csv") o_file = open("data/simbert_no_normal_vector.txt", "w", encoding="utf-8") for data in data_list: sentences = [data["标准问题*"]] vectors = model.encode(sentences, batch_size=64, convert_to_numpy=True, normalize_to_unit=False) # print(vectors) # vector = [str(float(i)) for i in vectors[0]] # print(vector) # o_file.writelines(sentences[0] + "--" + "--".join(vector) + "\n") o_file.close() except Exception as e: print(traceback.format_exc())