Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Simplify the usage of dict #2681

Merged
merged 4 commits into from
Sep 30, 2024
Merged

Simplify the usage of dict #2681

Changes from all commits
Commits
Show all changes
4 commits
Select commit Hold shift + click to select a range
687d398
Simplify the usage of dict
yqkcn Sep 30, 2024
e22aa11
fix
yqkcn Sep 30, 2024
abaa877
gix
yqkcn Sep 30, 2024
e63750b
Update graphrag/entity_resolution.py
KevinHuSh Sep 30, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
16 changes: 6 additions & 10 deletions graphrag/entity_resolution.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,7 @@
# See the License for the specific language governing permissions and
# limitations under the License.
#

import itertools
import logging
import re
import traceback
Expand Down Expand Up @@ -93,16 +93,12 @@ def __call__(self, graph: nx.Graph, prompt_variables: dict[str, Any] | None = No
node_clusters[graph.nodes[node]['entity_type']].append(node)

candidate_resolution = {entity_type: [] for entity_type in entity_types}
for node_cluster in node_clusters.items():
for k, v in node_clusters.items():
candidate_resolution_tmp = []
for a in node_cluster[1]:
for b in node_cluster[1]:
if a == b:
continue
if self.is_similarity(a, b) and (b, a) not in candidate_resolution_tmp:
candidate_resolution_tmp.append((a, b))
if candidate_resolution_tmp:
candidate_resolution[node_cluster[0]] = candidate_resolution_tmp
for a, b in itertools.permutations(v, 2):
if self.is_similarity(a, b) and (b, a) not in candidate_resolution_tmp:
candidate_resolution_tmp.append((a, b))
candidate_resolution[k] = candidate_resolution_tmp

gen_conf = {"temperature": 0.5}
resolution_result = set()
Expand Down