Hugging Face's logo

Spaces:
MicroPyscho
/
AidRenal
Sleeping

App Files Community
AidRenal / src /cnnClassifier /components /model_training.py

Commit History

  1. DenseNet121.keras modelre-run

    Okereke Kelechi Collins
    20fedd2

  1. ResNet50V2 model training

    Okereke Kelechi Collins
    643c37e

  2. model runs

    Okereke Kelechi Collins
    cb053fc

  1. running 2nd model

    Okereke Kelechi Collins
    36b8bfd

  1. new model run with 150 epochs and learning rate of 0.0003. Model checkpointing, early stopping and reduce learning rate on plateau callbacks have been added to the model training component. Model evaluation component has been updated to log validation loss and accuracy to mlflow and save it in scores.json file.

    Okereke Kelechi Collins
    ec0355f

  1. minor update

    Kelech Okereke Collins
    23761bd

  2. model_comparison

    Kelech Okereke Collins
    c246f66

  1. updates

    Kelech Okereke Collins
    d1f13c4