Skip to content
/ Tanoshi Public

A platform to train and test deep learning model using flask.

License

Notifications You must be signed in to change notification settings

rvk007/Tanoshi

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

75 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Tanoshi

progress badge Website

This is an end-to-end platform where you can upload your own custom dataset, set model parameters and train your own deep learning model without writing any line of code.



Tanoshi currently provides Image classification and Sentiment Analysis machine learning models. It has a feature to create a custom model by setting model parameters such as:

  • Batch size
  • Optimizer
  • Learning rate
  • Number of epochs
  • Training-Validation ratio

After validating the model parameters an input dataset file a user token is created and training starts, which takes around five-ten minutes. Once training is completed the user can use that token to test their model.

There are three major components of this project:

[Go to the above links to know more about this project in detail.]

The below image explains the work of each component and how they are related to each other:

Image Classification

Image classification, as the name suggests, ia an algorithm which predicts the content of an image. This project provides two different models, Resnet34 and MobileNetV2 which are pretrained on Imagenet dataset, to classify an image. Use the below format while creating the dataset and make sure to zip it before uploading else it won't be accepted.

Sentiment Analysis

Sentiment Analysis is a type of text classification, where a text is classified as Positive and Negative. Although, in this project Sentiment analysis is trained from scratch so it can be used for any kind text classification. Again, use the below specified format to create your dataset. The file should be a csv file.