Skip to content

Commit 894d988

Browse files
author
Francisco Santos
committed
removed unused n_feats argument
1 parent 1cebde4 commit 894d988

1 file changed

Lines changed: 1 addition & 3 deletions

File tree

examples/timeseries/tscwgan_example.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -26,16 +26,14 @@
2626
# Get transformed data stock - Univariate
2727
data, processed_data, scaler = processed_stock(path='./data/stock_data.csv', seq_len=seq_len, cols = 'Open')
2828
data_sample = processed_data[0]
29-
n_features = data_sample.shape[1]
3029

3130
model_parameters = ModelParameters(batch_size=batch_size,
3231
lr=learning_rate,
3332
betas=(beta_1, beta_2),
3433
noise_dim=noise_dim,
3534
n_cols=seq_len,
3635
layers_dim=dim,
37-
condition = cond_dim,
38-
n_features = n_features)
36+
condition = cond_dim)
3937

4038
train_args = TrainParameters(epochs=epochs,
4139
sample_interval=log_step,

0 commit comments

Comments
 (0)