19 lines
636 B
Python
19 lines
636 B
Python
import logging
|
|
|
|
from typing import Optional
|
|
|
|
from models.base_encoder import BaseEncoder
|
|
|
|
|
|
class BasicAutoEncoder(BaseEncoder):
|
|
# Based on https://medium.com/pytorch/implementing-an-autoencoder-in-pytorch-19baa22647d1
|
|
name = "BasicAutoEncoder"
|
|
|
|
def __init__(self, name: Optional[str] = None, input_shape: int = 0, loss_function=None):
|
|
self.log = logging.getLogger(self.__class__.__name__)
|
|
|
|
# Call superclass to initialize parameters.
|
|
super(BasicAutoEncoder, self).__init__(name, input_shape, loss_function)
|
|
|
|
# Network, optimizer and loss function are the same as defined in the base encoder.
|