mindsdb / lightwood

Lightwood is Legos for Machine Learning.
GNU General Public License v3.0
450 stars 94 forks source link

[ENH] Embedding mode #1163

Closed paxcema closed 1 year ago

paxcema commented 1 year ago

This feature enables training a Lightwood predictor without any mixers or subsequent analysis if the embedding_only flag is enabled in the problem definition. The resulting output is the concatenation of encoded representations learnt by each encoder, a.k.a. the "embedding".

Alternatively, normally trained predictors can also return these "embeddings" if the return_embedding argument is set to True.