mirror of
https://github.com/davrot/pytutorial.git
synced 2025-04-17 20:56:41 +02:00
Update README.md
Signed-off-by: David Rotermund <54365609+davrot@users.noreply.github.com>
This commit is contained in:
parent
7f55201d8b
commit
0866cdc4d8
1 changed files with 42 additions and 0 deletions
|
@ -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
|
||||
```
|
||||
|
||||
|
|
Loading…
Add table
Reference in a new issue