Skip to main content

Keras deep learning model implementations

Project description

# DeepModels
Implementations of various deep learning models using Keras.

## Requirements
An existing installation of either Tensorflow or Theano.

## Installation
```
pip install deep-models
```

## Usage

The models are implemented using Keras and instantiation returns a Keras Model object unless otherwise noted.

### Wide Residual Network
```python3
from deep_models import wide_residual_network as wrn

# Load your data
trainX = ...
trainY = ...
img_shape = (32, 32, 3)

# Create the model
# k is the width, 6 * n + 4 is the depth
model = wrn.build_model(img_shape, classes=10, n=4, k=10, dropout=0.3)

# Train the model
model.compile(loss="categorical_crossentropy", optimizer="adam", metrics=["acc"])
model.fit(
trainX, trainY,
batch_size=128,
epochs=100,
validation_split=0.2)
```

## Examples
Some working examples are available in the notebooks directory.

## License
See LICENSE file

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

Deep-Models-0.0.2.tar.gz (3.0 kB view hashes)

Uploaded Source

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page