Update baseline.py

Improve the algorithm, got 80.07 by score-segments.py