Add number of epochs as parameter
sparse experiment tunning