Automatically Generate Machine Learning Code with Just a Few Clicks

Traingenerator is a simple web application (deployed on Streamlit) that allows you to generate template codes in the two popular frameworks of PyTorch and scikit-learn. The generated code covers various segments expected of a code template, including preprocessing, model setup, training, and visualization. The templates available are currently limited to only image classification (with object detection on the way). Nonetheless, these boilerplate codes can be readily adapted to suit other ML tasks, and you can even add your own templates. The beauty of this app is that it simplifies the work of data scientists while lowering the barrier for non-technical users to explore ML codes.