Inspiration
I was looking at datasets on Kaggle and located a database that has every Ao3 tag and work in the year of 2021. This looked like a pretty cool dataset since I actively browse the website and wanted to locate statistics.
What it does
This model can create it's own tags on Ao3. As long as the word count is greater than 5000 and provided to the model, it can create tags semi-effectively.
How we built it
I used a K nearest neighbors model for my model.
Challenges we ran into
At first I tried to use a Multi label KNN model, but there were issues with locating the positional argument with the MKNN model, so I decided to use a KNN model instead, which does not have the error.
Accomplishments that we're proud of
I am very proud that I got to finish an AI model all by myself since previously I used mentor-guided datasets with specific models to use. This is the first time I had to choose what model I should use myself.
Built With
- kaggle
- python
- sklearn
Log in or sign up for Devpost to join the conversation.