TikTok Interview Question

try to implement logistic regression