forked from bhaycock/Siezure_Detection_Kaggle_Competition
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathtestPredictorMethod_H1_5spawn.py
More file actions
30 lines (25 loc) · 1.13 KB
/
Copy pathtestPredictorMethod_H1_5spawn.py
File metadata and controls
30 lines (25 loc) · 1.13 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
# Example quickrun to evaluate Epilepsy Data
# Making a standalone program that will read my data and play from there.
import epilepsyTools as eT
#import TestEstimators
import TransformsBH as transforms
import sys
from sklearn.ensemble import RandomForestClassifier
import numpy as np
import time
#import transform_Hills_cgv2B as Hill
listOfScans = ['Quick1']#, 'Dog_1', 'Dog_2', 'Dog_3', 'Dog_4', 'Dog_5', 'Patient_1', 'Patient_2']
#listOfDirectories = ['Dog_5', 'Patient_1', 'Patient_2']
#listOfPretransforms = [makeFive, genIctals]
#listOfTransforms = []
#listOfClassifiers = []
#listOfPosttransforms [fiveAllVote]
# Note- the fiveAllVote and makeFive are a lot simplier than I thought... mainly just train on the small segments as is, and then we separate the test sets to make the votes in the predictions.
def main():
directory = "/Users/BaZ/Desktop/KaggleTest/Data/"
name = "Hill1"
clf2 = RandomForestClassifier(n_estimators=3000, min_samples_split=1, bootstrap=False, n_jobs=4, random_state=0)
print eT.finalPredict(name, directory, listOfScans, transforms.HillTFWTFC, clf2, fiveSplit = True)
return
if __name__ == '__main__':
main()