diff --git a/demo/multi_classification/train.py b/demo/multi_classification/train.py index 2dc98f4d6..89a065450 100755 --- a/demo/multi_classification/train.py +++ b/demo/multi_classification/train.py @@ -1,4 +1,4 @@ - +#! /usr/bin/python import sys import numpy as np sys.path.append('../../python/') @@ -29,7 +29,6 @@ param['objective'] = 'multi:softmax' # scale weight of positive examples param['bst:eta'] = 0.1 param['bst:max_depth'] = 6 -param['eval_metric'] = 'auc' param['silent'] = 1 param['nthread'] = 4 param['num_class'] = 5