from sklearn.model selection import train _test_split

`from sklearn.model_selection import train_test_split`是一个用于将数据集分成训练集和测试集的函数。该函数可以将一个或多个数组作为输入，并根据指定的测试集大小或训练集大小将其拆分为两个不同的集合。以下是一个使用train_test_split函数的例子：

from sklearn.model_selection import train_test_split
import numpy as np

# 创建一个包含10个元素的数组
X = np.arange(10).reshape((5, 2))
y = range(5)

# 将数据集拆分为训练集和测试集
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.3, random_state=42)

# 输出拆分后的结果
print("X_train:\n", X_train)
print("X_test:\n", X_test)
print("y_train:\n", y_train)
print("y_test:\n", y_test)

该代码将一个包含10个元素的数组拆分为训练集和测试集，其中测试集大小为30%。输出结果将显示拆分后的训练集和测试集。