Esempio n. 1
0
        'NBG': naive_bayes.GaussianNB,
        'NBB': naive_bayes.BernoulliNB,
        "KNN": neighbors.KNeighborsClassifier,
        'ABC': ensemble.AdaBoostClassifier,
        'BGC': ensemble.BaggingClassifier,
        'GBC': ensemble.GradientBoostingClassifier,
        'RFC': ensemble.RandomForestClassifier,
        'ETC': ensemble.ExtraTreesClassifier
    }[model_id]()
    models.append((model, data_set))
    algorithms_list.append(model_id)

grid_search = True
## Set params
for model, feature_set in models:
    model.set_params(**classifier_utils.find_params(model, feature_set, trainX,
                                                    trainY, grid_search))

dates = {  #'Dates 1': [
    #{'training': ['1993-08-19', '2011-07-08'], 'testing': ['2011-07-11', '2016-04-20']},
    #{'training': ['1993-08-19', '2012-07-06'], 'testing': ['2012-07-09', '2016-04-20']},
    #{'training': ['1993-08-19', '2013-07-08'], 'testing': ['2013-07-09', '2016-04-20']}]
    'Dates 2': [
        {
            'training': ['1993-08-19', '2000-08-18'],
            'testing':
            ['2000-08-21', '2000-09-21', '2000-08-21', '2001-08-20']
        },
        {
            'training': ['1995-08-18', '2002-08-19'],
            'testing':
            ['2002-08-20', '2002-09-20', '2002-08-20', '2003-08-20']
             "KNN":neighbors.KNeighborsClassifier,

             'ABC': ensemble.AdaBoostClassifier,
             'BGC': ensemble.BaggingClassifier,             
             'GBC': ensemble.GradientBoostingClassifier,
             'RFC': ensemble.RandomForestClassifier,
             'ETC': ensemble.ExtraTreesClassifier        
            }[model_id]()
    models.append((model, data_set))
    algorithms_list.append(model_id)


grid_search = True
## Set params
for model, feature_set in models:
    model.set_params(**classifier_utils.find_params(model, feature_set, trainX, trainY, grid_search))


dates = {#'Dates 1': [
         #{'training': ['1993-08-19', '2011-07-08'], 'testing': ['2011-07-11', '2016-04-20']},
         #{'training': ['1993-08-19', '2012-07-06'], 'testing': ['2012-07-09', '2016-04-20']},
         #{'training': ['1993-08-19', '2013-07-08'], 'testing': ['2013-07-09', '2016-04-20']}]
         'Dates 2': [
          {'training': ['1993-08-19', '2000-08-18'], 'testing': ['2000-08-21', '2000-09-21', '2000-08-21', '2001-08-20']},
          {'training': ['1995-08-18', '2002-08-19'], 'testing': ['2002-08-20', '2002-09-20', '2002-08-20', '2003-08-20']},
#          {'training': ['1997-08-19', '2004-08-19'], 'testing': ['2004-08-20', '2004-09-20', '2004-08-20', '2005-08-19']},
#          {'training': ['1999-08-19', '2006-08-18'], 'testing': ['2006-08-21', '2006-09-20', '2006-08-21', '2007-08-20']},
#          {'training': ['2001-08-17', '2008-08-19'], 'testing': ['2008-08-20', '2008-09-22', '2008-08-20', '2009-08-20']},
#          {'training': ['2003-08-19', '2010-08-19'], 'testing': ['2010-08-19', '2010-09-20', '2010-08-19', '2011-08-19']},
#          {'training': ['2005-08-19', '2012-08-17'], 'testing': ['2012-08-20', '2012-09-20', '2012-08-20', '2013-08-20']},
          {'training': ['2007-08-17', '2015-05-19'], 'testing': ['2015-08-19', '2015-09-21', '2015-05-19', '2016-04-20']}]