Divisez l'ensemble de données (ndarray) en proportions arbitraires avec NumPy

Méthode

Utilisez numpy.split ().

(Exemple 1) Divisez l'ensemble de données en train: test = 7: 3

import numpy as np

ds = np.arange(128)  # array([0, 1, 2, ..., 127])

train, test = np.split(ds, [int(ds.size * 0.7)])

train  # array([[0, 1, ..., 88])
test   # array([[89, 90, ..., 127])

train.size  # 89 ≈ 128 * 0.7 = 89.6
test.size   # 39 ≈ 128 * 0.3 = 38.4

(Exemple 2) Divisez l'ensemble de données en train: test: validation = 6: 2: 2

import numpy as np

ds = np.arange(128)  # array([0, 1, 2, ..., 127])

indices = [int(ds.size * n) for n in [0.6, 0.6 + 0.2]]  # [76, 102]
train, test, validation = np.split(ds, indices)

train       # array([0, 1, ..., 75])
test        # array([76, 77, ..., 101])
validation  # array([102, 103, ..., 127])

train.size       # 76 ≈ 128 * 0.6 = 76.8
test.size        # 26 ≈ 128 * 0.2 = 25.6
validation.size  # 26 ≈ 128 * 0.2 = 25.6

référence

Recommended Posts

Divisez l'ensemble de données (ndarray) en proportions arbitraires avec NumPy
Petite histoire: divisez l'alphabet en nombres arbitraires et sortie (résolu)
Divisez les données en unités de type projet avec Django (2)
[GIMP] [Python-Fu] Divisez l'image en épées
Divisez les données en unités de type projet avec Django (3)
Trouvez la position au-dessus du seuil avec NumPy
Divisez la chaîne de caractères en le nombre de caractères spécifié
Divisez les données en unités de type projet avec Django