AI controlled electrical brain stimulation