不能让“bert”模型使用ktrain和熊猫数据

Question

我试着和ktrain一起完成bert模型的制作。我使用名为train_df的熊猫数据存储我的数据。

x_train, x_val, y_train, y_val = train_test_split(train_df['text'], train_df['target'], shuffle=True, test_size = 0.2, random_state=random_seed, stratify=train_df['target'])

​

我使用函数texts_from_array，因为当我想将数据转换为ValueError (ValueError: x_train必须是list或NumPy数组)时，我使用的是熊猫数据。

(x_train_bert,  y_train_bert), (x_val_bert, y_val_bert), preproc = text.texts_from_array(x_train=x_train, y_train=y_train,                                                                                         x_test = x_val, y_test=y_val,
                                                                                          class_names= ["0", "1"],
                                                                                          preprocess_mode='bert',
                                                                                          lang = 'en',
                                                                                          maxlen=65,
                                                                                          max_features=35000)

我错过了什么？

Answer 1

我找到了解决方案，现在它工作正常了。

(x_train_bert, y_train_bert), (x_val_bert, y_val_bert), preproc = text.texts_from_array(x_train=x_train.tolist(), y_train=y_train.tolist(), x_test = x_val.tolist(), y_test=y_val.tolist(),class_names= ["0", "1"],preprocess_mode='bert',lang = 'en', maxlen=65, max_features=35000)