from .model_helpers import make_tensorboard_callback, make_save_path
from ..utils import naming


class SimpleModel:
    def __init__(self, directory_name: str, n_input: int, n_output: int):
        self.directory_name = directory_name

        self.model = keras.models.Sequential()
        self.model.add(keras.layers.Dense(400, input_dim=n_input))
        self.model.add(keras.layers.Activation('relu'))

        self.model.add(keras.layers.Dense(800))
        self.model.add(keras.layers.Activation('relu'))