Skip to content

Commit

Permalink
changed learning rates
Browse files Browse the repository at this point in the history
  • Loading branch information
aagrawl3 committed Dec 14, 2016
1 parent d5247c0 commit 154ff2c
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion lambdamart.py
Original file line number Diff line number Diff line change
Expand Up @@ -91,7 +91,7 @@ def get_pairs(scores):

class LambdaMART:

def __init__(self, training_data=None, number_of_trees=0, learning_rate=0, tree_type='sklearn'):
def __init__(self, training_data=None, number_of_trees=5, learning_rate=0.1, tree_type='sklearn'):
'''
The format for training data is as follows:
[relevance, q_id, [feature vector]]
Expand Down

0 comments on commit 154ff2c

Please sign in to comment.