Metadata-Version: 2.1
Name: airontools
Version: 0.1.23
Summary: Machine learning tools to complement the AIronSuit package.
Home-page: https://github.com/AtrejuArtax/airontools
Author: Claudi Ruiz Camps
Author-email: claudi_ruiz@hotmail.com
License: BSD
Classifier: Programming Language :: Python :: 3
Classifier: License :: OSI Approved :: BSD License
Classifier: Operating System :: OS Independent
Description-Content-Type: text/markdown
License-File: LICENSE
Requires-Dist: absl-py (==1.4.0)
Requires-Dist: array-record (==0.4.0)
Requires-Dist: astunparse (==1.6.3)
Requires-Dist: attrs (==23.1.0)
Requires-Dist: cachetools (==5.3.1)
Requires-Dist: certifi (==2023.5.7)
Requires-Dist: charset-normalizer (==3.1.0)
Requires-Dist: click (==8.1.3)
Requires-Dist: dm-tree (==0.1.8)
Requires-Dist: etils (==1.3.0)
Requires-Dist: flatbuffers (==23.5.26)
Requires-Dist: gast (==0.4.0)
Requires-Dist: google-auth (==2.17.3)
Requires-Dist: google-auth-oauthlib (==1.0.0)
Requires-Dist: google-pasta (==0.2.0)
Requires-Dist: googleapis-common-protos (==1.59.1)
Requires-Dist: grpcio (==1.56.0)
Requires-Dist: h5py (==3.9.0)
Requires-Dist: idna (==3.4)
Requires-Dist: importlib-metadata (==6.7.0)
Requires-Dist: importlib-resources (==5.12.0)
Requires-Dist: jax (==0.4.13)
Requires-Dist: joblib (==1.2.0)
Requires-Dist: keras (==2.12.0)
Requires-Dist: libclang (==16.0.0)
Requires-Dist: Markdown (==3.4.3)
Requires-Dist: MarkupSafe (==2.1.3)
Requires-Dist: ml-dtypes (==0.2.0)
Requires-Dist: more-itertools (==9.1.0)
Requires-Dist: numpy (==1.23.5)
Requires-Dist: oauthlib (==3.2.2)
Requires-Dist: opt-einsum (==3.3.0)
Requires-Dist: packaging (==23.1)
Requires-Dist: pluggy (==0.13.1)
Requires-Dist: promise (==2.3)
Requires-Dist: protobuf (==4.23.3)
Requires-Dist: psutil (==5.9.5)
Requires-Dist: py (==1.11.0)
Requires-Dist: pyasn1 (==0.5.0)
Requires-Dist: pyasn1-modules (==0.3.0)
Requires-Dist: pytest (==5.4.3)
Requires-Dist: requests (==2.31.0)
Requires-Dist: requests-oauthlib (==1.3.1)
Requires-Dist: rsa (==4.9)
Requires-Dist: scikit-learn (==1.2.2)
Requires-Dist: scipy (==1.10.1)
Requires-Dist: six (==1.16.0)
Requires-Dist: tensorboard (==2.12.3)
Requires-Dist: tensorboard-data-server (==0.7.1)
Requires-Dist: tensorflow-datasets (==4.9.2)
Requires-Dist: tensorflow-estimator (==2.12.0)
Requires-Dist: tensorflow-macos (==2.12.0)
Requires-Dist: tensorflow-metadata (==1.13.1)
Requires-Dist: tensorflow-metal (==0.8.0)
Requires-Dist: termcolor (==2.3.0)
Requires-Dist: threadpoolctl (==3.1.0)
Requires-Dist: toml (==0.10.2)
Requires-Dist: tqdm (==4.65.0)
Requires-Dist: typing-extensions (==4.6.3)
Requires-Dist: urllib3 (==2.0.3)
Requires-Dist: wcwidth (==0.2.6)
Requires-Dist: Werkzeug (==2.3.6)
Requires-Dist: wrapt (==1.14.1)
Requires-Dist: zipp (==3.15.0)

# AIronTools

AIronTools (Beta) is a Python library that provides the user with deep learning tools built to work with 
[tensorflow](https://github.com/tensorflow/tensorflow) as a backend.

Key features:

1. Model constructor that allows multiple models to be optimized in parallel across multiple GPUs. 
2. Block constructor to build customised blocks/models.
3. Layer constructor to build customised layers.
4. Preprocessing utils.
   
### Installation

`pip install airontools`

### Custom Keras subclass to build a variational autoencoder (VAE) with airontools and compatible with aironsuit

``` python
import numpy as np
from numpy.random import normal
from tensorflow.keras.optimizers import Adam

from airontools.constructors.models.unsupervised.vae import VAE


tabular_data = np.concatenate(
    [
        normal(loc=0.5, scale=1, size=(100, 10)),
        normal(loc=-0.5, scale=1, size=(100, 10)),
    ]
)
model = VAE(
        input_shape=tabular_data.shape[1:],
        latent_dim=3,
)
model.compile(optimizer=Adam(learning_rate=0.001))
model.fit(
    tabular_data,
    epochs=10,
)
print("VAE evaluation:", float(model.evaluate(tabular_data)["loss"]))
```

### More examples

see usage examples in [aironsuit/examples](https://github.com/AtrejuArtax/aironsuit/tree/master/examples)
