add study.py to do some tries using a sample of the training dataset