Ai like scikit-learn for Lua

Hi!
I’m thinking about making a simple Ai based game in Defold. But I’m not sure what Ai library I should use. I’m looking for something like the TreeClassifier in scikit-learn for Python. You give it a list of floats with features, and then a list with labels. Then to make a prediction you give it some features and it will predict the label. Is there anything like that for Lua that I can use in Defold?
Thanks in forehand!
Br Imim