safebooru-predict
Predict user ratings for anime-style images
Getting Started
- Install the required python packages with
pip install -r requirements.txt - Download the data and place
all_data.csvintores/safebooru/data
Generate statistics
Enter the src/preprocess directory and run python preprocess_all_data.py
Cost: 2.5 GB RAM, 16 minutes
Find correlations between tags
Enter the src/correlate directory and run python compute_cooccurrence_matrix.py
Cost: 5.5 GB RAM, 35 minutes
Run python query.py and explore tags.