IBM Bob 2.0 hackathon team: ML-dive