Add option to set training test fraction @ cmd line