__init__.py 851 Bytes
Newer Older
1 2
#from DataShuffler import *
from .Layer import Layer
3
from .Conv1D import Conv1D
4 5 6
from .Conv2D import Conv2D
from .FullyConnected import FullyConnected
from .MaxPooling import MaxPooling
Tiago de Freitas Pereira's avatar
Tiago de Freitas Pereira committed
7
from .AveragePooling import AveragePooling
8
from .Dropout import Dropout
Tiago de Freitas Pereira's avatar
Tiago de Freitas Pereira committed
9
from .InputLayer import InputLayer
Tiago de Freitas Pereira's avatar
Scratch  
Tiago de Freitas Pereira committed
10

Tiago de Freitas Pereira's avatar
Tiago de Freitas Pereira committed
11

12
# gets sphinx autodoc done right - don't remove it
Tiago de Freitas Pereira's avatar
Tiago de Freitas Pereira committed
13 14 15 16 17 18 19 20 21 22 23 24 25 26 27
def __appropriate__(*args):
  """Says object was actually declared here, an not on the import module.

  Parameters:

    *args: An iterable of objects to modify

  Resolves `Sphinx referencing issues
  <https://github.com/sphinx-doc/sphinx/issues/3048>`
  """

  for obj in args: obj.__module__ = __name__

__appropriate__(
    Layer,
28
    Conv1D,
Tiago de Freitas Pereira's avatar
Tiago de Freitas Pereira committed
29 30 31 32 33
    Conv2D,
    FullyConnected,
    MaxPooling,
    AveragePooling,
    Dropout,
34
    InputLayer,
Tiago de Freitas Pereira's avatar
Tiago de Freitas Pereira committed
35
    )
36
__all__ = [_ for _ in dir() if not _.startswith('_')]
Tiago de Freitas Pereira's avatar
Scratch  
Tiago de Freitas Pereira committed
37