Added trainer for linear embedding