Correct the trainset
Setup the weights
Dimensions should match the input documents, unsuprisingly