-
Notifications
You must be signed in to change notification settings - Fork 105
Description
when running this code on 4.basic_GAN.py
if name == 'main':
input_dim = X_train.shape[1]
feature_size = X_train.shape[2]
output_dim = y_train.shape[1]
For Bayesian
opt = {"lr": 0.00016, "epoch": 165, 'bs': 128}
generator = make_generator_model(X_train.shape[1], output_dim, X_train.shape[2])
discriminator = make_discriminator_model()
gan = GAN(generator, discriminator, opt)
Predicted_price, Real_price, RMSPE = gan.train(X_train, y_train, yc_train, opt)
error showing
AttributeError Traceback (most recent call last)
~\AppData\Local\Temp/ipykernel_1848/1524428345.py in
4 discriminator = make_discriminator_model()
5 gan = GAN(generator, discriminator, opt)
----> 6 Predicted_price, Real_price, RMSPE = gan.train(X_train, y_train, yc_train, opt)
AttributeError: 'GAN' object has no attribute 'train'