Skip to content

Latest commit

 

History

History
9 lines (5 loc) · 473 Bytes

README.md

File metadata and controls

9 lines (5 loc) · 473 Bytes

Movie Genre Guessing ML Model (@ Resnet152 & FastAI + Pytorch)

This is a computer vision classification model trained using FastAI and PyTorch using the resnet152 framework. It classifies between 5 genres viz: Horror, Action, Fantasy, Animation, and Romance.

The training process included: Preprocessing -> Training -> Cleaning the data -> Training -> Export

You can test it @ HuggingFace Space