qualia_core.learningmodel.keras package

Submodules

Module contents

qualia_core.learningmodel.keras.CNN(input_shape, output_shape, filters: tuple = (6, 16, 120), kernel_sizes: tuple = (3, 3, 5), paddings: tuple = (0, 0, 0), batch_norm: bool = False, dropouts: float | list[float] = 0.0, pool_sizes: tuple = (2, 2, 0), fc_units: int = (84,), prepool: int = 0, dims: int = 1)[source]
qualia_core.learningmodel.keras.MLP(input_shape, output_shape, units: tuple = (100, 100, 100))[source]
qualia_core.learningmodel.keras.ResNet(input_shape, output_shape, filters: int = (15, 15, 30, 60, 120), kernel_sizes: tuple = (3, 3, 3, 3, 3), num_blocks: tuple = (2, 2, 2, 2), strides: tuple = (1, 1, 2, 2, 2), paddings: int = (1, 1, 1, 1, 1), prepool: int = 1, batch_norm: bool = False, bn_momentum: float = 0.99, force_projection_with_stride: bool = True, dims: int = 1)[source]