Skip to content

jfouts2010/MTGCardColorPredictor-ML

Repository files navigation

MTGCardColorPredictor-ML

A model that predicts the color of the card based on the first 50 words of the card text with keras. It achieved around an 85% accuracey rating during testing, which is great for how simple the model is. The model is a simple DNN with 3 deep layers and dropout, binary cross entropy for loss (because multilabel classification) and Adam optimizer.

To create the model, run ModelCreation.py. To test Kaldheim Cards (Most recent set and model did not train on the set), run ModelEvaluate.py. To test specific text, run ModelEvaluateTest.py, changing the text in the file.

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages