Fixed reddit example (#132)

Raphtory · Apr 5, 2023 · 2c06a16 · 2c06a16
1 parent 56570e1
commit 2c06a16
Show file tree

Hide file tree

Showing 3 changed files with 389 additions and 91 deletions.
diff --git a/docbrown/db/src/graph_loader/reddit_hyperlinks.rs b/docbrown/db/src/graph_loader/reddit_hyperlinks.rs
@@ -1,5 +1,5 @@
 //! Load (a subset of) Reddit hyperlinks dataset into a graph.
-//! The dataset is available at https://snap.stanford.edu/data/soc-redditHyperlinks-body.tsv
+//! The dataset is available at http://snap.stanford.edu/data/soc-redditHyperlinks-title.tsv
 //! The hyperlink network represents the directed connections between two subreddits (a subreddit
 //! is a community on Reddit). We also provide subreddit embeddings. The network is extracted
 //! from publicly available Reddit data of 2.5 years from Jan 2014 to April 2017.
@@ -55,8 +55,8 @@ use std::path::PathBuf;
 /// * `PathBuf` - The path to the file
 pub fn reddit_file(timeout: u64) -> Result<PathBuf, Box<dyn std::error::Error>> {
     fetch_file(
-        "reddit.tsv",
-        "https://snap.stanford.edu/data/soc-redditHyperlinks-body.tsv",
+        "reddit-title.tsv",
+        "http://snap.stanford.edu/data/soc-redditHyperlinks-title.tsv",
         timeout,
     )
 }
@@ -111,15 +111,15 @@ pub fn reddit_graph(shards: usize, timeout: u64) -> Graph {
                                     ("readability".to_string(), Prop::F64(post_properties[17])),
                                     (
                                         "positive_sentiment".to_string(),
-                                        Prop::F64(post_properties[17]),
+                                        Prop::F64(post_properties[18]),
                                     ),
                                     (
                                         "negative_sentiment".to_string(),
-                                        Prop::F64(post_properties[17]),
+                                        Prop::F64(post_properties[19]),
                                     ),
                                     (
                                         "compound_sentiment".to_string(),
-                                        Prop::F64(post_properties[17]),
+                                        Prop::F64(post_properties[20]),
                                     ),
                                 ];
                                 g.add_vertex(time, *src_id, &vec![])

diff --git a/python/tests/demo.ipynb b/python/tests/demo.ipynb
diff --git a/python/tests/reddit_example.ipynb b/python/tests/reddit_example.ipynb