Skip to content

tune your model through web app, instantly stop and retrain your model (applicable with phone!)

Notifications You must be signed in to change notification settings

zehuichen123/ModelTuner

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Model Tuner

Intro

This is a web app which enables you to tune your model through PC and Mobile simutaneously. It is powered by Flask and Vue.js. Back-end is responsible for handle model trials and results and front-end provides interface to start/stop your model with different config and visualize trial results.

Framework

Some key point if you want to apply this framework to your work:

  • Choose your matric for your model and write them into to one .csv file at each print time where can be read by flask.

  • Expose your model trial with one interface say start_trial().

    num_runs = 10
    def start_trial(config):
        data_config = load_config() # return value is a dict
        data_config.update(config)
        for i in range(num_runs):
            model.run(data)
  • Call your start_trial function at back_end/app/run.py.

  • Customize your format of results shown in the front-end and rewrite function data(),table() and class Data in back_end/app/run.py. You may also need to change some files in Vue.js project. :)

So actually this project probably used by myself. If you want your own model tuner, I strongly suggest you write one on your own. It will only take you one night :).

About

The whole framework is quite simple since the motivation of this application is to enable me to choose different parameters and check results and then choose again on the phone. I don't need to sit in front of table and watch my screen: epoch 1, epoch 2, epoch ....

About

tune your model through web app, instantly stop and retrain your model (applicable with phone!)

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published