huggingface_hub
30516923 - Keras model card (#679)

Commit
3 years ago
Keras model card (#679) * Initial commit * directory path fix * fix with os.path * fix with os.path * added hyperparameters and other parts of card * readme fix * quality * added model summary * summary fix * summary fix * added model history to model card * history fix * model plot added * history fix * swapped model tag with library name keras * yaml metadata fix * yaml metadata reverted * bug fix * toggle fix * remove toggle for summary * increment in epochs * put image inside toggle * fixed invalid yaml metadata * plot path fix * plot path fix * plot path fix * added additional task_name tag for people to tag the model * tests * Update src/huggingface_hub/keras_mixin.py Co-authored-by: Omar Sanseviero <osanseviero@gmail.com> * changed require tf decorator for unit test * I didn't add file_download lol * added graphviz and pydot to deps in setup.py * added graphviz and pydot to deps in setup.cfg * removed two redundant variables, carried functions for model card creation in push_to_hub_keras() and other nits * fixed dependency condition * added graphviz installation * fitting the model for model history in push to hub function tests * added checking readme and model image from siblings in model info * fixed condition on require_tf * fighting with graphviz brb * added plotting as a separate parameter and added both pip and apt pkg of graphviz * added logging back * make style * Update src/huggingface_hub/keras_mixin.py Co-authored-by: Omar Sanseviero <osanseviero@gmail.com> * Update src/huggingface_hub/keras_mixin.py Co-authored-by: Omar Sanseviero <osanseviero@gmail.com> * added functions under create_model_card, added test for model plotting false, other nits * small fix * Update src/huggingface_hub/keras_mixin.py Co-authored-by: Omar Sanseviero <osanseviero@gmail.com> * removed two arguments from tests, put history parsing under write model card * added tests for pushing built model, edited model card for built and built and compiled model, cleaned tests * make style * changed model plotting variable name * model.history.history type fix * save_pretrained_keras with model card and plotting * swapped one test case from push_to_hub_keras to save_pretrained_keras * make style Co-authored-by: Omar Sanseviero <osanseviero@gmail.com>
Author
Parents
Loading