一些 AutoGluon 用户在使用 TextPredictor.fit() 函数进行文本分类时可能会碰到一个错误:BrokenPipeError: [Errno 32] Broken pipe。这个错误可能是因为训练数据太大而导致网络断开连接。为了解决这个问题,你可以尝试使用以下代码将训练数据分成更小的数据块:
import pandas as pd
from autogluon.text import TextPredictor
train_data = pd.read_csv('train.csv')
train_data_split = train_data.iloc[:1000] # Split train data into smaller chunks
predictor = TextPredictor(label='label', problem_type='multiclass')
predictor.fit(train_data_split)
在上述示例中,我们将 train_data 数据集切分成了前 1000 行。你也可以根据你的电脑性能和数据集大小自己决定切分数据的行数。
上一篇:AutogluonTabularPredictorfailswithTypeError:stringindicesmustbeintegers
下一篇:Autogluon中出现“AttributeError:'TabularDataset'objecthasnoattribute'unique'”错误。