dataset_total = pd.concat((dataset_train, dataset_test), axis = 0)