Skip to content

[Solved] RuntimeError: Input and parameter tensors are not at the same device, found input tensor at cpu and parameter tensor at cuda:0

[Scikit-Learn] Using "train_test_split()" to split your data

Last Updated on 2021-06-28 by Clay

Today, if we need to split our data for training our model —— we need to split training data and test data. We use "training data" to train our model and check it never peep our "test data", it is very important, because our "test data" can assess the quality of our model.

Read More »[Scikit-Learn] Using "train_test_split()" to split your data
Exit mobile version