python - 機械学習におけるデータのテストとトレーニング

Question

ANN 用に 2000 以上のデータ セットがあります。その中に MLPRegressor を適用しました。私のコードは正常に動作しています。しかし、テストのために、たとえば 50 個のデータ セットがあるなどのテスト値を修正したいと考えています。それから、最初の 20 の値をテストしたいと思います。コードでこれを修正するにはどうすればよいですか? 次のコードを使用しました。

import pandas as pd

import matplotlib.pyplot as plt

from sklearn.neural_network import MLPRegressor

df = pd.read_csv("0.5-1.csv")
df.head()

X = df[['wavelength', 'phase velocity']]
y = df['shear wave velocity']

from sklearn.model_selection import train_test_split

X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=.2)

from sklearn.neural_network import MLPClassifier
from sklearn.metrics import mean_absolute_error

mlp = MLPRegressor(hidden_layer_sizes=(30,30,30))

mlp.fit(X_train,y_train)