Keras 프레임워크 아래의 단독 인코딩과 디코딩으로 당신의 데이터를 대변신시킵니다

1495 단어 기계 학습
이 슬라이드에서는 2D 행렬을 다룹니다.
# -*- coding: utf-8 -*-
import numpy as np
from keras.utils import to_categorical


def _OneHot_encode():
    data = np.array([[0, 1, 2],
                     [3, 4, 5],
                     [7, 8, 9],
                     [10, 11, 12]])
    print(data)
    print(data.shape)
    encoded_data = to_categorical(data)
    print(encoded_data.shape)
    print(encoded_data)
    return encoded_data


def _OneHot_decode():
    encoded_data = _OneHot_encode()
    decoded_data = []
    for i in range(encoded_data.shape[0]):
        decoded = np.argmax(encoded_data[i], axis=1)
        decoded_data.append(decoded)
    decoded_data = np.array(decoded_data)
    print(decoded_data.shape)
    return decoded_data


if __name__ == '__main__':
    decoded_data = _OneHot_decode()
    print(decoded_data)

여기서 인코딩을 마치면 다음과 같습니다.
[[[1. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0.]   [0. 1. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0.]   [0. 0. 1. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0.]]
 [[0. 0. 0. 1. 0. 0. 0. 0. 0. 0. 0. 0. 0.]   [0. 0. 0. 0. 1. 0. 0. 0. 0. 0. 0. 0. 0.]   [0. 0. 0. 0. 0. 1. 0. 0. 0. 0. 0. 0. 0.]]
 [[0. 0. 0. 0. 0. 0. 0. 1. 0. 0. 0. 0. 0.]   [0. 0. 0. 0. 0. 0. 0. 0. 1. 0. 0. 0. 0.]   [0. 0. 0. 0. 0. 0. 0. 0. 0. 1. 0. 0. 0.]]
 [[0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 1. 0. 0.]   [0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 1. 0.]   [0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 1.]]]
그리고 디코딩을 마치고 돌아갔습니다.
[[ 0  1  2]  [ 3  4  5]  [ 7  8  9]  [10 11 12]]

좋은 웹페이지 즐겨찾기