Skip to content

Commit

Permalink
Adding README file
Browse files Browse the repository at this point in the history
  • Loading branch information
avkhimen committed Apr 2, 2022
1 parent 04bc7f3 commit 192be2c
Show file tree
Hide file tree
Showing 4 changed files with 34 additions and 4 deletions.
2 changes: 0 additions & 2 deletions .ipynb_checkpoints/cwgan-gp_time_series-checkpoint.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -11,11 +11,9 @@
"from tensorflow import keras\n",
"from tensorflow.keras import layers\n",
"\n",
"#from tensorflow_docs.vis import embed\n",
"import matplotlib.pyplot as plt\n",
"import tensorflow as tf\n",
"import numpy as np\n",
"import imageio\n",
"plt.rcParams['figure.figsize'] = (20.0, 8.0)"
]
},
Expand Down
31 changes: 31 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,31 @@
# Conditional Wasserstein GAN for Synthetic Time Series Generation

Conditional Wasserstein GAN with gradient penalty for the generation of synthetic time series.

## Description

Propriatary dataset was used to train the conditional WGAN with gradient penalty. Any dataset with shape (num_samples, num_features) will work.

Run the cells in sequence in `cwgan-gp_time_series.ipynb` jupyter notebook. Final cell contains code to create synthetic
image conditioned on a label.

Code in part based on:

* https://keras.io/examples/generative/conditional_gan/
* https://keras.io/examples/generative/wgan_gp/
* https://keras.io/examples/generative/dcgan_overriding_train_step/

## Getting Started

### Dependencies

* tensorflow
* numpy
* matplotlib

See `requirements.txt` file.


## License

Free to use for any purpose
2 changes: 0 additions & 2 deletions cwgan-gp_time_series.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -11,11 +11,9 @@
"from tensorflow import keras\n",
"from tensorflow.keras import layers\n",
"\n",
"#from tensorflow_docs.vis import embed\n",
"import matplotlib.pyplot as plt\n",
"import tensorflow as tf\n",
"import numpy as np\n",
"import imageio\n",
"plt.rcParams['figure.figsize'] = (20.0, 8.0)"
]
},
Expand Down
3 changes: 3 additions & 0 deletions requirements.txt
Original file line number Diff line number Diff line change
@@ -0,0 +1,3 @@
tensorflow
numpy
matplotlib

0 comments on commit 192be2c

Please sign in to comment.