-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathxgboosts.py
More file actions
73 lines (43 loc) · 1.59 KB
/
xgboosts.py
File metadata and controls
73 lines (43 loc) · 1.59 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
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
from numpy import loadtxt
import numpy as np
import tensorflow as tf
import pandas as pd
import math
from xgboost import XGBClassifier
from xgboost import XGBRegressor
from sklearn.model_selection import train_test_split
from sklearn.metrics import accuracy_score
path = "/home/andrea/Desktop/python/tensorflowTutorialGenerical/"
abalone_train = path + "training"
abalone_test = path + "test"
abalone_toPredict = path + "toPredict"
CSV_COLUMNS = [ "a","b","c","d","e","f","g","target"]
# Training examples
dataset_train = pd.read_csv(abalone_train, names=CSV_COLUMNS, skipinitialspace=True)
dataset_test = pd.read_csv(abalone_test, names=CSV_COLUMNS, skipinitialspace=True)
dataset_toPredict = pd.read_csv(abalone_toPredict, names=CSV_COLUMNS, skipinitialspace=True)
#print (dataset)
X_train = dataset_train[["a","b","c","d","e","f","g"]]
Y_train = dataset_train["target"]
X_test = dataset_test[["a","b","c","d","e","f","g"]]
Y_test = dataset_test["target"]
X_toPredict = dataset_toPredict[["a","b","c","d","e","f","g"]]
Y_toPredict = dataset_toPredict["target"]
#print(Y)
# fit model no training data
model = XGBRegressor()
model.fit(X_train, Y_train)
print( model )
y_pred = model.predict( X_test )
y_pred_brief = model.predict( X_toPredict )
#print( y_pred )
MSE = 0
for i in range( 0, len(Y_test) ):
toAdd = (Y_test[i] - y_pred[i])**2
toAdd = math.sqrt( toAdd )
MSE = MSE + toAdd
MSE = MSE / len(Y_test)
print("MSE : " + str(MSE) )
for i in range ( 0 , len(Y_toPredict) ):
print( "expected: " + str( y_pred_brief[i] ) + "; vs real: " + str( Y_toPredict[i] ) )
print("finished")