Calculate Cost Function In Python