Skip to content

Latest commit

 

History

History
29 lines (20 loc) · 1.12 KB

README.md

File metadata and controls

29 lines (20 loc) · 1.12 KB
Raw

NLP Challenge for Core42 AI Modelling team

Goal

Build a multi class image classification model using any framework(PyTorch, Keras, etc..) and deploy via a REST API

Format Requirements

  • Fork or clone this repo
  • Maintain the directory structure given for the project
  • Use Python 3.7+
  • If you need additional imports specify them in requirements.txt

Model Requirements

  • Model should be trained on the given dataset.
  • Create a model artifact and save it under /models.
  • Report accuracy on test.

API Requirements

  • Serve the model as a REST API using FastAPI
  • Be able to use CURL to send in text input and return the prediction.

Data

The dataset you will be using contains 10 classes of apparels. You need to train basic image classification model which will classify given text into these 10 classes categories. You can find the dataset and details data format and labels can be found here

Dataset contains two splits train and test.

Submission

Timebox this challenge to 8-10 hours. After completing the assignment, please compress whole repo and send it.