Python Evaluator.TrainAndSaveAlgorithm Beispiele

Beispiel #1

def BuildRecModel():
    np.random.seed(0)
    random.seed(0)

    # Load up common data set for the recommender algorithms
    (evaluationData, rankings) = LoadData()
    # Construct an Evaluator to, you know, evaluate them
    evaluator = Evaluator(evaluationData, rankings)

    hotelKNNAlgorithm = HotelKNNAlgorithm()
    log.info('Algorithm=K-Nearest Neighbour')
    evaluator.SetAlgorithm(hotelKNNAlgorithm, 'HotelKNNAlgorithm')

    evaluator.Evaluate(False)
    evaluator.TrainAndSaveAlgorithm()