Create ML

add training/testing module