Update src/train2ban.py

Updated predict method to use reshaped ip_feature_db