Skip to content

Commit

Permalink
adding dining preference dataset (#3866)
Browse files Browse the repository at this point in the history
This dataset is very small, and uses strings as node names. It will be used to test force atlas, and in a new link prediction/similarity notebook.

the licensing is contained here.
http://networkdata.ics.uci.edu/netdata/html/Dining-table_partners.html

Authors:
  - Don Acosta (https://github.com/acostadon)
  - Brad Rees (https://github.com/BradReesWork)

Approvers:
  - Brad Rees (https://github.com/BradReesWork)
  - ralph (https://github.com/nv-rliu)

URL: #3866
  • Loading branch information
acostadon authored Sep 21, 2023
1 parent db5073d commit a7047e3
Show file tree
Hide file tree
Showing 3 changed files with 27 additions and 0 deletions.
1 change: 1 addition & 0 deletions python/cugraph/cugraph/datasets/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -27,6 +27,7 @@
meta_path = Path(__file__).parent / "metadata"

cyber = Dataset(meta_path / "cyber.yaml")
dining_prefs = Dataset(meta_path / "dining_prefs.yaml")
dolphins = Dataset(meta_path / "dolphins.yaml")
email_Eu_core = Dataset(meta_path / "email_Eu_core.yaml")
karate = Dataset(meta_path / "karate.yaml")
Expand Down
23 changes: 23 additions & 0 deletions python/cugraph/cugraph/datasets/metadata/dining_prefs.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,23 @@
name: dining_prefs
file_type: .csv
description: Classic social networking dataset describes dining preferences for a dormitory in New York state.
author: J.L. Moreno
refs:
J. L. Moreno (1960). The Sociometry Reader. The Free Press, Glencoe, Illinois, pg.35
delim: " "
header: None
col_names:
- src
- dst
- wgt
col_types:
- string
- string
- int
has_loop: false
is_directed: false
is_multigraph: false
is_symmetric: true
number_of_edges: 42
number_of_nodes: 26
url: https://data.rapids.ai/cugraph/datasets/dining_prefs.csv
3 changes: 3 additions & 0 deletions python/cugraph/cugraph/testing/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -23,6 +23,7 @@
)
from cugraph.datasets import (
cyber,
dining_prefs,
dolphins,
karate,
karate_disjoint,
Expand All @@ -42,6 +43,7 @@
UNDIRECTED_DATASETS = [karate, dolphins]
SMALL_DATASETS = [karate, dolphins, polbooks]
WEIGHTED_DATASETS = [
dining_prefs,
dolphins,
karate,
karate_disjoint,
Expand All @@ -51,6 +53,7 @@
small_tree,
]
ALL_DATASETS = [
dining_prefs,
dolphins,
karate,
karate_disjoint,
Expand Down

0 comments on commit a7047e3

Please sign in to comment.