コード例 #1
0
def train(xTrain, yTrain, metric):
    print 'RandomForestClassifier'
    global forest
    forest = RandomForestClassifier()
    forest.fit(xTrain, yTrain)
    global trainResults
    trainResults = forest.predict(xTrain)
    i.setSuccess(trainResults, metric)
コード例 #2
0
ファイル: gradientBoosting.py プロジェクト: lramsey/higgs
def train(xTrain, yTrain, metric):
    print 'goosting'
    global gboost
    gboost = GBC()
    gboost.fit(xTrain,yTrain)
    global trainResults
    trainResults = gboost.predict_proba(xTrain)[:,1]
    i.setSuccess(trainResults, metric)
コード例 #3
0
ファイル: adaboost.py プロジェクト: lramsey/higgs
def train(xTrain, yTrain, metric):
    print 'adaboost'
    global boost
    boost = AdaBoostClassifier()
    boost.fit(xTrain,yTrain)
    global trainResults
    trainResults = boost.predict_proba(xTrain)[:,1]
    i.setSuccess(trainResults, metric)
コード例 #4
0
ファイル: adaboost.py プロジェクト: ismailezzaki96/HiggsMl
def train(xTrain, yTrain, metric):
    print 'adaboost'
    global boost
    boost = AdaBoostClassifier()
    boost.fit(xTrain, yTrain)
    global trainResults
    trainResults = boost.predict_proba(xTrain)[:, 1]
    i.setSuccess(trainResults, metric)
コード例 #5
0
ファイル: svm.py プロジェクト: lramsey/higgs
def train(xTrain, yTrain, metric):
    print 'svm'
    global vector
    vector = SVC()
    vector.fit(xTrain, yTrain)
    global trainResults
    trainResults = vector.predict(xTrain)
    i.setSuccess(trainResults, metric)
コード例 #6
0
ファイル: logistic.py プロジェクト: ismailezzaki96/HiggsMl
def train(xTrain, yTrain, metric):
    print 'logistic'
    global logis
    logis = LogisticRegression()
    logis.fit(xTrain, yTrain)
    global trainResults
    trainResults = logis.predict_proba(xTrain)[:, 1]
    i.setSuccess(trainResults, metric)
コード例 #7
0
def train(xTrain, yTrain, metric):
    print 'svm'
    global vector
    vector = SVC()
    vector.fit(xTrain, yTrain)
    global trainResults
    trainResults = vector.predict(xTrain)
    i.setSuccess(trainResults, metric)
コード例 #8
0
def train(xTrain, yTrain, metric):
    print 'goosting'
    global gboost
    gboost = GBC()
    gboost.fit(xTrain, yTrain)
    global trainResults
    trainResults = gboost.predict_proba(xTrain)[:, 1]
    i.setSuccess(trainResults, metric)
コード例 #9
0
ファイル: logistic.py プロジェクト: lramsey/higgs
def train(xTrain, yTrain, metric):
    print 'logistic'
    global logis
    logis = LogisticRegression()
    logis.fit(xTrain,yTrain)
    global trainResults
    trainResults = logis.predict_proba(xTrain)[:,1]
    i.setSuccess(trainResults, metric)