StockMarket-Prediction/main_code.py at main · Intelligence08/StockMarket-Prediction · GitHub

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
31
32
33
34
35
36
import python_ie221 as p1
import pickle

#Read data
dt = p1.ReadData('data\Combined_News_DJIA.csv')

#Preprocessing
pre = p1.PreProcessing()

pre.fully_preprocess(dt.data)

#Processing
process = p1.Processing(pre.x_train, pre.x_test, pre.y_train, pre.y_test)
process.RandomForest()
process.KNNClassifier()
process.Naive()
process.SVMLinearSVC()

#Result
result = p1.Result(process)
result.full_score()

##visualize
visual = p1.Visualization()
visual.result_visualization(pre.pre_data, result.List_score)

#Save class contains models by pickle
filename = "Model"
outfile = open(filename,'wb')
pickle.dump(process,outfile)
outfile.close()

#Load class contains models by pickle(Optional)
#infile = open(filename,'rb')
#class_model = pickle.load(infile)
#infile.close()