tensorly / viz

Easy visualization and evaluation of matrix and tensor factorization models
https://tensorly.org/viz/
MIT License
16 stars 4 forks source link

Factor tools for PARAFAC2 #21

Open cchatzis opened 3 months ago

cchatzis commented 3 months ago

Hi there, thank you making my life easier with tlviz. I was wondering, would there be interest in updates of functions in factor_tools.py to support PARAFAC2Tensors?

yngvem commented 2 months ago

Hi @cchatzis, thank you for the kind feedback! 😄

When we created TLViz, Marie and I explicitly made the choice to focus on PARAFAC (and small amount of Tucker) to keep the scope and matinainer work low. However, we also did our best to keep it general enough to make it possible to use TLViz with other decompositions.

We don't have any examples showcasing this in the example gallery though, so one thing that would be very valuable is an example for the gallery that demonstrates how to turn PARAFAC2Tensor into a "CPTensor" form to use the functions in factor_tools for evaluating them. Maybe even how to use multimodal_evaluation with custom similarity metrics and error functions.

cchatzis commented 2 months ago

@yngvem I currently work mainly with PARAFAC2Tensors and tlviz, so I definetely see the value in that. Expect a PR in the future :)