step23. 패키지로 정리

📢 본 포스팅은 밑바닥부터 시작하는 딥러닝3을 기반으로 작성하였습니다. 배운 내용을 기록하고, 개인적인 공부를 위해 작성하는 포스팅입니다. 자세한 내용은 교재 구매를 강력 추천드립니다.

 

 

이번 단계에서는 지금까지의 코드를 패키지로 정리하겠습니다. dezero 디렉터리를 만들고 모듈들을 추가합니다. 파이썬 파일은 모듈이 되고, 모듈들을 디렉터리에 모으면 패키지가 됩니다. 이렇게 만들어지는 dezero 패키지가 현재 만들고 있는 프레임워크가 되는 것입니다. 우선은 core_simple.py라는 파일을 만들고 step22의 클래스와 함수들을 그대로 복사합니다. 나중에는 최종 형태인 core.py로 교체하게 될 것입니다.

 

코어파일인 dezero/core_simple.py에 연산자들을 오버로드해주는 함수를 추가합니다.

def setup_variable():
    Variable.__add__ = add
    Variable.__radd__ = add
    Variable.__mul__ = mul
    Variable.__rmul__ = mul
    Variable.__neg__ = neg
    Variable.__sub__ = sub
    Variable.__rsub__ = rsub
    Variable.__truediv__ = div
    Variable.__rtruediv__ = rdiv
    Variable.__pow__ = pow

 

앞선 함수들은 dezero/__init__.py 파일에서 호출하도록 하겠습니다. 이 파일은 모듈을 import할 때 가장 먼저 실행됩니다. 이때 33단계에서부터는 core.py로 대체할 예정이므로 core_simple.py와 core.py 중 하나를 선택하여 import 할 수 있도록 설정합니다. 우선 core.py를 import하는 부분은 비워두도록 하겠습니다.

is_simple_core = True

if is_simple_core:
    from dezero.core_simple import Variable
    from dezero.core_simple import Function
    from dezero.core_simple import using_config
    from dezero.core_simple import no_grad
    from dezero.core_simple import as_array
    from dezero.core_simple import as_variable
    from dezero.core_simple import setup_variable
else:
    pass

setup_variable()

 

이제 다음과 같이 dezero 패키지를 활용할 수 있습니다. 검색 경로 추가 코드는 dezero 디렉터리를 import 하기 위해 일시적으로 사용합니다. 후에 pip install 등으로 DeZero가 패키지로 설치된다면, 파이썬 검색 경로에 추가되기 때문에 수동으로 경로를 추가하지 않아도 됩니다. 앞으로는 dezero 디렉터리에 있는 모듈들을 확장하는 방식으로 진행됩니다.

# Add import path for the dezero directory
if '__file__' in globals():
    import os, sys
    sys.path.append(os.path.join(os.path.dirname(__file__), '..'))

import numpy as np
from dezero import Variable

x = Variable(np.array(1.0))
y = (x + 3) ** 2
y.backward()

print(y)  # variable(16.0
print(x.grad)  # 8.0