[console_scripts]
pgse-predict = pgse.cli:predict
pgse-train = pgse.cli:train
