made training/testing sets but cannot predict