diff --git a/scikit-learn/svm/README.md b/scikit-learn/svm/README.md index e72a5f1..ecc92d6 100644 --- a/scikit-learn/svm/README.md +++ b/scikit-learn/svm/README.md @@ -54,6 +54,48 @@ predict(X) > > Class labels for samples in X. +## Test and train data + +```python +import numpy as np + +rng = np.random.default_rng(1) + +a_x: np.ndarray = rng.normal(1.5, 1.0, size=(1000))[:, np.newaxis] +a_y: np.ndarray = rng.normal(3.0, 1.0, size=(1000))[:, np.newaxis] +data_train_0: np.ndarray = np.concatenate((a_x, a_y), axis=-1) +class_train_0: np.ndarray = np.full((data_train_0.shape[0],), -1) + +a_x: np.ndarray = rng.normal(1.5, 1.0, size=(1000))[:, np.newaxis] +a_y: np.ndarray = rng.normal(3.0, 1.0, size=(1000))[:, np.newaxis] +data_test_0: np.ndarray = np.concatenate((a_x, a_y), axis=-1) +class_test_0: np.ndarray = np.full((data_test_0.shape[0],), -1) +del a_x +del a_y + +a_x: np.ndarray = rng.normal(0.0, 1.0, size=(1000))[:, np.newaxis] +a_y: np.ndarray = rng.normal(0.0, 1.0, size=(1000))[:, np.newaxis] +data_train_1: np.ndarray = np.concatenate((a_x, a_y), axis=-1) +class_train_1: np.ndarray = np.full((data_train_0.shape[0],), +1) + +a_x: np.ndarray = rng.normal(0.0, 1.0, size=(1000))[:, np.newaxis] +a_y: np.ndarray = rng.normal(0.0, 1.0, size=(1000))[:, np.newaxis] +data_test_1: np.ndarray = np.concatenate((a_x, a_y), axis=-1) +class_test_1: np.ndarray = np.full((data_test_0.shape[0],), +1) +del a_x +del a_y + +data_train: np.ndarray = np.concatenate((data_train_0, data_train_1), axis=0) +data_test: np.ndarray = np.concatenate((data_test_0, data_test_1), axis=0) +label_train: np.ndarray = np.concatenate((class_train_0, class_train_1), axis=0) +label_test: np.ndarray = np.concatenate((class_test_0, class_test_1), axis=0) + +np.save("data_train.npy", data_train) +np.save("data_test.npy", data_test) +np.save("label_train.npy", label_train) +np.save("label_test.npy", label_test) +``` + ```python ```