oujago/NumpyDL

View on GitHub
npdl/layers/__init__.py

Summary

Maintainability
A
0 mins
Test Coverage
# -*- coding: utf-8 -*-


# import file
from . import base
from . import convolution
from . import core
from . import embedding
from . import normalization
from . import pooling
from . import recurrent
from . import shape


# base
from .base import Layer


# common layers
from .core import Linear
from .core import Dense
from .core import Softmax
from .core import Dropout


# embedding layer
from .embedding import Embedding


# recurrent layers
from .recurrent import Recurrent
from .recurrent import SimpleRNN
from .recurrent import GRU
from .recurrent import LSTM
from .recurrent import BatchLSTM


# convolution
from .convolution import Convolution


# shape layer
from .shape import Flatten
from .shape import DimShuffle


# pooling layers
from .pooling import MaxPooling
from .pooling import MeanPooling


# normalization layer
from .normalization import BatchNormal