import xgboost as xgb
import pandas as pd
import numpy as np
from PIL import Image

# load data
img_obj = Image.open('bob_digit/5/aaaa0_x0_y2.bmp')
data_list = list((255 - np.array(img_obj)).reshape(-1))

# load bst model
bst_model = xgb.Booster({'bthread': 2})
bst_model.load_model('bob_digit/bob_digit_classification.model')

df = pd.DataFrame()
col_names = ['f%d' % i for i in range(680)]
df = df.append(pd.Series(data=data_list, index=col_names), ignore_index=True)

result1 = bst_model.predict(xgb.DMatrix(df))
print(np.around(result1[0], decimals=2))

np_arr = np.zeros(shape=(1, 680))
np_arr[0] = data_list

result2 = bst_model.predict(xgb.DMatrix(np_arr))
print(np.around(result2[0], decimals=2))