diff --git a/diary-training.py b/diary-training.py index 05212196368b7245810ba20adda0524293efd6c3..58646d002390fd32910434e845e5a8124dfdcb1d 100644 --- a/diary-training.py +++ b/diary-training.py @@ -166,7 +166,7 @@ ffm_model.setTrain("/home/zhangyanzhao/data.csv") param = {'task':'binary', 'lr':0.05, 'lambda':0.002, 'metric':'auc', 'fold':3} -ffm_model.cv(param, '/home/zhangyanzhao/model.out') +ffm_model.cv(param) ''' ffm_model.setTest("/home/zhangyanzhao/test.csv")