Skip to content
Alexander L. Hayes edited this page Dec 28, 2016 · 10 revisions

Dataset 1: "IMDB"

All Datasets: boost-starai/BoostSRL-Datasets

by: Nandini Ramanan, Alexander L. Hayes

Previous Page | BoostSRL Wiki | Next Page: "Cora Dataset"


Table of Contents:

  1. Overview
  2. Download
  3. Setup
  4. Modes

Overview

Internet Movie Database (IMDB) is an online database of movies, television shows, the actors that star in them, and the people that make them.

Target:

  • female_gender

Table of Contents - BoostSRL Wiki


Download

Download: IMDB.zip (29.6 KB)

  • md5sum: 62f2b9936664fde55b6eb2ac05f87619

  • sha256sum: 7b42fd0f563236735885ae474dc19791c456af12989fc0667f598d25f0dba57a

Table of Contents - BoostSRL Wiki


Setup

Linux/Mac:

  1. After downloading, unzip IMDB.zip

unzip IMDB.zip

  1. If you're using a jar file, move it into the CiteSeer directory:

mv (jar file) IMDB/

  1. Learning:
  • java -jar BoostSRL.jar -l -train train/ -target female_gender -trees 10
  1. Inference:
  • java -jar BoostSRL.jar -i -test test/ -models train/models/ -target female_gender -trees 10

Windows:

(Coming soon)

Table of Contents - BoostSRL Wiki


Modes

setParam: treeDepth=3.
setParam: nodeSize=3.
setParam: numOfClauses=8.
setParam: numOfCycles=8.
setParam: recursion=true. 
setParam: lineSearch=true.
setParam: resampleNegs=true.
mode: actor(+person).
mode: recursive_actor(`person).
mode: recursive_movie(+movie, `person).
mode: recursive_movie(`movie, +person).
mode: movie(+movie, +person).
mode: movie(+movie, -person).
mode: movie(-movie, +person).
mode: female_gender(+person).
mode: recursive_female_gender(`person).
mode: genre(+person, +genre).
mode: genre(+person, #genre).
mode: genre(+person, -genre).
mode: genre(-person, +genre).
mode: recursive_genre(+person, `genre).
mode: recursive_genre(`person, +genre).
mode: workedunder(+person, +person).
mode: workedunder(+person, -person).
mode: workedunder(-person, +person).
mode: recursive_workedunder(+person, `person).
mode: recursive_workedunder(`person, +person).
okIfUnknown: recursive_actor/1.
okIfUnknown: recursive_movie/2.
okIfUnknown: recursive_female_gender/1.
okIfUnknown: recursive_genre/2.
okIfUnknown: recursive_workedunder/2.
okIfUnknown: movie_2_1_genre/3.
okIfUnknown: workedunder_2_1_genre/3.
okIfUnknown: workedunder_1_1_gender/2.
okIfUnknown: movie_2_1_gender/2.
usePrologVariables: true.
mode: movie_2_1_genre(+movie, +genre, #count).
mode: movie_2_1_genre(+movie, #genre, #count).
mode: movie_2_1_gender(+movie, #count).
mode: workedunder_2_1_genre(+person, +genre, #count).
mode: workedunder_2_1_genre(+person, #genre, #count).
mode: workedunder_1_1_gender(+person, #count).
mode: movie_1(+person, #count).
mode: movie_2(+movie, #count).
mode: workedunder_1(+person, #count).
mode: workedunder_2(+person, #count).
precompute:
movie_2_1_genre(M, G, C) :- movie(M,P1), genre(P1,G), all(P, movie(M, P)^genre(P,G), AllP), C is length(AllP).
precompute: 
movie_2_1_gender(M, C) :- movie(M,P1), female_gender(P1), all(P, movie(M, P)^female_gender(P), AllP), C is length(AllP).
precompute: 
workedunder_2_1_genre(M, G, C) :- workedunder(M,P1), genre(P1,G), all(P, workedunder(M, P)^genre(P,G), AllP), C is length(AllP).
precompute: 
workedunder_1_1_gender(M, C) :- workedunder(P1,M), female_gender(P1), all(P, workedunder(P, M)^female_gender(P), AllP), C is length(AllP).
precompute:
movie_1(P,C) :- movie(M,P), all(M, movie(M,P), AllM), C is length(AllM).
precompute:
movie_2(M,C) :- movie(M,P), all(P, movie(M,P), AllM), C is length(AllM).
precompute:
workedunder_1(D, C) :- workedunder(A,D), all(A, workedunder(A,D), AllA), C is length(AllA).
precompute:
workedunder_2(A, C) :- workedunder(A,D), all(D, workedunder(A,D), AllD), C is length(AllD).

Table of Contents - BoostSRL Wiki


Previous Page | BoostSRL Wiki | Next Page: "Cora Dataset"