Java Deep Learning Projects

Movie rating prediction

Now that all of the datasets required for training, validating, and evaluating are ready, we can start training the FM model. We first start by showing the filename for the training data:

final String trainFile = formattedDataPath+ "/" + "ratings_train.libfm";

Then, we set the testing data file path:

final String testFile = formattedDataPath+ "/" + "ratings_test.libfm";

Then, we set the testing metadata file path:

final String testMetaFile = formattedDataPath+ "/" + "ratings_test.libfm.meta";

Then, the filename for the final prediction output file:

final String outputFile = formattedDataPath+ "/" + "predict_output.txt";

Then, we set up the path for writing the logs, metrics, time, and so on for each iteration to a ...