From 19e03696015e05e92de3a56c33cf7d6f7fdbcc47 Mon Sep 17 00:00:00 2001 From: Carlota de la Vega Date: Tue, 18 Jun 2024 11:40:26 +0200 Subject: [PATCH] Pylint corrections --- src/drawing.py | 3 ++- src/nums/cgan.py | 2 +- src/nums/training.py | 7 ++++--- 3 files changed, 7 insertions(+), 5 deletions(-) diff --git a/src/drawing.py b/src/drawing.py index 6d1aad7..8e78ffe 100644 --- a/src/drawing.py +++ b/src/drawing.py @@ -28,6 +28,7 @@ def draw_number(number, cond_gan): return generated_image + def draw_image(description, cond_gan): noise = np.random.normal(size=(1, utils.latent_dim)) label = keras.utils.to_categorical([description], builders.num_classes) @@ -47,4 +48,4 @@ def draw_image(description, cond_gan): filename = f"./data/images/drawn_image_{description}.png" imageio.imwrite(filename, generated_image) - return generated_image \ No newline at end of file + return generated_image diff --git a/src/nums/cgan.py b/src/nums/cgan.py index d66e10f..bdd32b3 100644 --- a/src/nums/cgan.py +++ b/src/nums/cgan.py @@ -13,7 +13,7 @@ def __init__(self, discriminator, generator, latent_dim, image_size, num_classes self.num_classes = num_classes -# pylint: disable=too-many-ancestors, too-many-instance-attributes +# pylint: disable=too-many-ancestors, too-many-instance-attributes, too-many-locals class ConditionalGAN(keras.Model): def __init__( self, config, d_optimizer, g_optimizer, loss_fn diff --git a/src/nums/training.py b/src/nums/training.py index 7686bc9..8db3160 100644 --- a/src/nums/training.py +++ b/src/nums/training.py @@ -1,7 +1,8 @@ from src import utils +from src.nums import builders -dataset = utils.load_dataset() -generator, discriminator = utils.build_models() -cond_gan = utils.build_conditional_gan(generator, discriminator) +dataset = utils.load_dataset("mnist") +generator, discriminator = builders.build_models() +cond_gan = builders.build_conditional_gan(generator, discriminator) utils.train_model(dataset, cond_gan) utils.save_model_weights(cond_gan, "cond_weights.weights.h5")