Code for network architectures.
Module Contents
-
class
SanityCheckNetwork[source]
Bases: tensorflow.python.keras.Sequential
A network consisting of a single fully connected layer.
-
__init__(self)[source]
-
class
SimpleCubeCnn[source]
Bases: tensorflow.python.keras.Sequential
A simple 3D CNN for TESS data cubes.
-
__init__(self)[source]
-
class
SimpleLightcurveCnn[source]
Bases: tensorflow.python.keras.Model
A simple 1D CNN for lightcurves.
-
__init__(self)[source]
-
call(self, inputs, training=False, mask=None)[source]
-
class
SimpleLightcurveLstm[source]
Bases: tensorflow.python.keras.Model
-
__init__(self)[source]
-
call(self, inputs, training=False, mask=None)[source]
-
class
SimpleLightcurveCnnPerTimeStepLabel[source]
Bases: tensorflow.python.keras.Model
A simple 1D CNN for lightcurves.
-
__init__(self)[source]
-
call(self, inputs, training=False, mask=None)[source]
-
class
Conv1DTranspose(filters, kernel_size, strides=1, *args, **kwargs)[source]
Bases: tensorflow_core.python.keras.engine.Layer
-
__init__(self, filters, kernel_size, strides=1, *args, **kwargs)[source]
-
build(self, input_shape)[source]
-
call(self, x, training=False, mask=None)[source]
-
compute_output_shape(self, input_shape)[source]
-
class
ConvolutionalLstm[source]
Bases: tensorflow.python.keras.Model
-
__init__(self)[source]
-
call(self, inputs, training=False, mask=None)[source]