It's an Implementation of ANN with callbacks
Project description
OneFlow
It is a Package having the Implementation of ANN with callbacks
🔗 Project Link
Check out the Pypi Package here
Run Locally
Create two files in your working directory:
- config.yaml
- training.py
config.yaml
params:
epochs : 3
batch_size : 32
num_classes : 10
input_shape : [28, 28]
loss_function : sparse_categorical_crossentropy
metrics : accuracy
optimizer : SGD
validation_datasize : 5000
es_patience : 5
artifacts:
artifacts_dir : artifacts
model_dir : model
plots_dir : plots
model_name : model.h5
plot_name : results_plot.png
model_ckpt_dir : ModelCheckpoints
callbacked_model_name : model_ckpt.h5
logs:
logs_dir : logs_dir
general_logs : general_logs
tensorboard_root_log_dir : tensorboard_logs
training.py
from OneFlow.utils.common import read_config
from OneFlow.utils.data_mgmt import get_data
from OneFlow.utils.model import StepFlow
import argparse, os
def training(config_path):
config = read_config(config_path)
validation_datasize = config["params"]["validation_datasize"]
#This "get_data" function is loading the mnist dataset, bring your own and divide into categories to perform the custom training
(X_train, y_train), (X_valid, y_valid), (X_test, y_test) = get_data(validation_datasize)
sp = StepFlow(config, X_train, y_train, X_valid, y_valid)
sp.create_model()
sp.fit_model()
sp.save_final_model()
sp.save_plot()
if __name__ == "__main__":
args = argparse.ArgumentParser()
args.add_argument("-c", "--config", default="config.yaml")
parsed_args = args.parse_args()
training(config_path = parsed_args.config)
Then run the following commands on the termial
pip install OneFlow-Hassi34
python training.py
On completion of training, run the following command on termial and observe the metrics on tensorboard
tensorboard --logdir=logs_dir/tensorboard_logs/
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
OneFlow-Hassi34-0.0.7.tar.gz
(17.7 kB
view hashes)
Built Distribution
Close
Hashes for OneFlow_Hassi34-0.0.7-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | de66b703c227f225f4fe3f58e2fe39af200950dff7030bbedc3937c37b06e48d |
|
MD5 | d2f3af3179902411fba57e9ee8cae47f |
|
BLAKE2b-256 | cf16c876ccecbd3fb0de314e83fc3e2bb09bbde83fa961247ee9603c771a4d2e |