谈谈wustl EE硕士转CS还是Data

我是一位wustl EE的研1硕士，因为我们学校有点特别，Data专业是设置在EE学院下面的，所以不管是选课啊，EE➕data双学位都是很通融的，但是因为我本科也是EE的，感觉是不是数学的基础没有科班学统计或者数学的同学强？(本科什么线代微积分概率论啥的基本课还都是学过的）本科EE学的代码知识也是少之又少.....所以感觉EE的定位很尴尬.
我现在在选一些machine learning，optimization的课，感觉还能跟的上，但是看了大多数EE的同学都在往CS的方向转，又有点担心是不是自己又走错方向了.求走过来的大神指点迷津！