Transpose the trainset and the testset and intergrate the iteration option.
Use the right command line parameters to run the method.
Add the interation option.
Integrate the iteration option and remove unnecessary options.