Skip to content

HTTPS clone URL

Subversion checkout URL

You can clone with HTTPS or Subversion.

Download ZIP
branch: master

Fetching latest commit…

Cannot retrieve the latest commit at this time

Failed to load latest commit information.
java
preprocess
README.md

README.md

This repository contains code to support the work described in Bamman, O'Connor and Smith, "Learning Latent Personas of Film Characters" (ACL 2013), including:

  1. Generating model input data from Stanford CoreNLP-processed XML files and movie/character metadata from Freebase.
  2. Training a persona model on this data.

1. GENERATE INPUT DATA (optional)

This step is provided to document the process of text conversion so that the models below can be run on new, different texts. If you want run the models on existing movie data (found in java/input/movies.data), this step can be skipped. Follow the instructions in pipeline.sh to run the preprocessing pipeline to generate input data.

cd preprocess

./pipeline.sh (Takes several hours)

2. TRAIN MODELS

This step trains a new persona model. The run.sh script has a number of variables that can be set, including the number of latent personas (A), the number of latent topics (K), etc. Here, INPUT is the path to movie data (either the default located in java/input/movies.data or the output from step 1 above). OUTPUT_DIRECTORY is the location to write all output files to.

cd ../java

gunzip input/movies.data.gz

./run.sh $OUTPUT_DIRECTORY $INPUT (Takes several hours)

Something went wrong with that request. Please try again.