مدل شبکه عصبی (یادگیری عمیق) برای پیش بینی کلاس در دیتاست

روی دیتاست با لینک زیر: https://drive.google.com/file/d/1EgbTmPFPyEmc8fgPY-1I2PtXdysob-8q/view?usp=sharing میخواهم با یادگیری عمیق یا هر روشی که میدونید بهتر جواب میده، پیش بینی کلاس رو داشته باشم. دو ستون آخر، دو کلاس یا برچسب رو نشون میده. حتما برای تست نتایج، test از validation و train جدا باشد در ابتدای کار در داده های آموزش و fit مدل، از test به هیچ وجه استفاده نشود تا نتایج واقعی باشد. حجم دیتاست 90M، تعداد سطرها، 372184، تعداد ستون های ویژگی ها 18، تعداد ستون های برچسب 2