我正在尝试使用Fast_ml将我的数据拆分为训练、验证和测试集，以用于机器学习目的。我的输入和输出数据都是通过np.load从.npy文件中读取的。输入”P”是一个形状为(100000, 4, 4, 6, 1)的数组，而目标”Q”是一个形状为(100000,)的向量。我使用了以下代码：

from fast_ml.model_development import train_valid_test_splitX_train, y_train, X_valid, y_valid, X_test, y_test = train_valid_test_split(P, Q,                                                                             train_size=0.8,                                                                            valid_size=0.1,                                                                            test_size=0.1)

然而，我收到了以下错误：

AttributeError: 'numpy.ndarray' object has no attribute 'drop'

回答：

这个问题通过以下方法解决了：

from sklearn.model_selection import train_test_splitX_train, X_rem, y_train, y_rem = train_test_split(P,Q, train_size=0.8)X_valid, X_test, y_valid, y_test = train_test_split(X_rem,y_rem, test_size=0.5)