# OrGAN

GAN-based generation of organization structures.

It is based on the ideas of MolGAN: An implicit generative model for small molecular graphs (https://arxiv.org/abs/1805.11973)

This implementation is based on
* [yongqyu/MolGAN-pytorch](https://github.com/yongqyu/MolGAN-pytorch)
* [nicola-decao/MolGAN](https://github.com/nicola-decao/MolGAN)

Please note, that in the base repository (by yongqyu) there is an open issue that the results are not like in the MolGAN paper (official implementation is by nicola-decao.


## Dependencies

* **python>=3.5**
* **pytorch>=0.4.1**: https://pytorch.org
* **numpy**

## Structure
* `data`: should contain organization structure dataset. You can use `generate_dataset` script to generate one.

## Usage
```
python main.py
```