Beispiel #1
0
'''
This implements the standard SGD algorithm
'''

from params import Params
from latent_model import LatentModel
from mm_data_file import MMDataFile
from sgd_mf_machine import SGDMachine
import time
import numpy as np


if __name__ == "__main__":
    # Parameters/Model initialization
    params = Params()
    params.parse_args("SGD method.")
    params.print_params()

    model = LatentModel(params)

    # Handling I/O things
    data = MMDataFile()
    data.read_file(params)
    print "File load successful..."
    
    # Run the actual training program
    program = SGDMachine()
    start = time.clock()
    program.train(params, model, data)
    proc_time = (time.clock() - start)