support --test-predict option