topic Re: Cosine similarity between all rows pairwise on a dataset of 100million rows in Data Engineering

Cosine similarity between all rows pairwise on a dataset of 100million rows

Databricks2005 — Thu, 16 Sep 2021 21:08:13 GMT

Hello everyone,

I am facing performance issue while calculating cosine similarity in pyspark on a dataframe with around 100 million records.

I am trying to do a cross self join on the dataframe to calculate it.

The executors are all having same number of tasks when seen on the spark ui.

The input size to all executors is also almost the same.

Executors : 20

Cores: 4 cores

Any inputs would be highly appreciated

Re: Cosine similarity between all rows pairwise on a dataset of 100million rows

Databricks2005 — Fri, 17 Sep 2021 09:35:20 GMT

Thank you Kaniz !. I shall wait for an answer 🙂

Re: Cosine similarity between all rows pairwise on a dataset of 100million rows

john_odwyer — Wed, 22 Sep 2021 21:41:16 GMT

The issue is probably related to the self join between 100 million rows, I'm not positive without seeing the code and understanding the problem better but you may want to think about using windowing functions instead

https://blog.knoldus.com/using-windows-in-spark-to-avoid-joins/

Re: Cosine similarity between all rows pairwise on a dataset of 100million rows

Sonal — Fri, 01 Oct 2021 17:00:26 GMT

Is there a way to hash the record attributes so that the cartesian join can be avoided? I work on record similarity and fuzzy matching and we do a learning based blocking alorithm which hashes the records into smaller buckets and then the hashes are joined. You can check https://github.com/zinggAI/zingg for the approach.