updated tests, improved prediction speed#35

SubhadityaMukherjee merged 2 commits intomasteropenml/openml-pytorch:masterfrom

fixing_tests_and_validationopenml/openml-pytorch:fixing_tests_and_validationCopy head branch name to clipboard