minor changes

Author: domkowald

src/common/Bookmark.java

@@ -37,17 +37,21 @@ public class Bookmark implements Comparable<Bookmark> {
 	private String timestamp;
 	private List<Integer> tags;
 
-	private double rating;
 	private List<Integer> categories;
 
+	private double rating;
+	private String title;
+	
 	public Bookmark(int userID, int wikiID, String timestamp) {
 		this.userID = userID;
 		this.resID = wikiID;
 		this.timestamp = timestamp;
 
-		this.rating = -2.0;
 		this.categories = new ArrayList<Integer>();
 		this.tags = new ArrayList<Integer>();
+		
+		this.rating = -2.0;
+		this.title = null;
 	}
 
 	@Override
@@ -99,6 +103,14 @@ public void setRating(double rating) {
 		this.rating = rating;
 	}
 
+	public String getTitle() {
+		return this.title;
+	}
+	
+	public void setTitle(String title) {
+		this.title = title;
+	}
+	
 	public List<Integer> getCategories() {	
 		return this.categories;
 	}

src/file/BookmarkReader.java

@@ -131,6 +131,11 @@ public boolean readFile(String filename) {
 
 				// TODO ----------------------
 				// extend common/Bookmark class with fields for title (= lineParts[6]) and description (= lineParts[7])
+				if (lineParts.length > 6) { // is there a rating?
+					try {
+						userData.setTitle(lineParts[5].replace("\"", ""));
+					} catch (Exception e) { /* do nothing */ }
+				}
 			}
 			processUserData(userID, userData, tags, categories, wikiID); // last user
 			br.close();

src/file/BookmarkSplitter.java

@@ -220,6 +220,13 @@ public static boolean writeSample(BookmarkReader reader, List<Bookmark> userSamp
 				bw.write("\"");
 				if (bookmark.getRating() != -2) {
 					bw.write(";\"" + bookmark.getRating() + "\"");
+				} else {
+					bw.write("\"\"");
+				}
+				if (bookmark.getTitle() != null) {
+					bw.write(";\"" + bookmark.getTitle() + "\"");
+				} else {
+					bw.write("\"\"");
 				}
 				bw.write("\n");
 			}

src/test/Pipeline.java

@@ -87,8 +87,8 @@ public class Pipeline {
 	// placeholder for the topic posfix
 	private static String TOPIC_NAME = null;
 	// placeholder for the used dataset
-	private final static String DATASET = "del";
-	private final static String SUBDIR = "/ipm3";
+	private final static String DATASET = "twitter";
+	private final static String SUBDIR = "";//"/ipm3";
 
 	public static void main(String[] args) {
 		System.out.println("TagRecommender:\n" + "" +
@@ -113,14 +113,14 @@ public static void main(String[] args) {
 		//TagReuseProbAnalyzer.analyzeSample(path, TRAIN_SIZE, TEST_SIZE, false);
 
 		//evaluate(dir, path, "pitf", TOPIC_NAME, true, true, null);
-		//try { getStatistics("bib_core/vedran/bib_bibtex_2_perc_1", false); } catch (Exception e) { e.printStackTrace(); }
+		//try { getStatistics("twitter_core/seedUserTweets", false); } catch (Exception e) { e.printStackTrace(); }
 
 		//JSONProcessor.writeJSONOutput("bib_core/vedran/bib_bibtex_2_perc_1");
 
 		//evaluateAllTagRecommenderApproaches(dir, path);
 		//startAllTagRecommenderApproaches(dir, path, true);
 		//getTrainTestStatistics(path);
-		//BookmarkSplitter.splitSample(DATASET + "_core/twitter", DATASET + "_core/" + DATASET + "_sample", 1, 0, true);
+		//BookmarkSplitter.splitSample("twitter_core/seedUserTweets", "twitter_core/twitter_sample", 1, 0, true);
 		//BookmarkSplitter.drawUserPercentageSample("bib_core/vedran/bib_bibtex", 5);
 		//createLdaSamples("ml_core/core1/ml_sample", 1, 1000, true, true);