the-algorithm/twml/twml/layers/__init__.py
twitter-team ef4c5eb65e Twitter Recommendation Algorithm
Please note we have force-pushed a new initial commit in order to remove some publicly-available Twitter user information. Note that this process may be required in the future.
2023-03-31 17:36:31 -05:00

22 lines
1.0 KiB
Python

# pylint: disable=wildcard-import
"""
This module contains the ``tf.layers.Layer`` subclasses implemented in twml.
Layers are used to instantiate common subgraphs.
Typically, these layers are used when defining a ``build_graph_fn``
for the ``twml.trainers.Trainer``.
"""
from .batch_prediction_tensor_writer import BatchPredictionTensorWriter # noqa: F401
from .batch_prediction_writer import BatchPredictionWriter # noqa: F401
from .data_record_tensor_writer import DataRecordTensorWriter # noqa: F401
from .full_dense import full_dense, FullDense # noqa: F401
from .full_sparse import full_sparse, FullSparse # noqa: F401
from .isotonic import Isotonic # noqa: F401
from .layer import Layer # noqa: F401
from .mdl import MDL # noqa: F401
from .partition import Partition # noqa: F401
from .percentile_discretizer import PercentileDiscretizer # noqa: F401
from .sequential import Sequential # noqa: F401
from .sparse_max_norm import MaxNorm, sparse_max_norm, SparseMaxNorm # noqa: F401
from .stitch import Stitch # noqa: F401