MA-sLDAr -- Multi-Annotator Supervised LDA for regression
MA-sLDAr is a C++ implementation of the supervised topic models with response variables provided by multiple annotators with different levels of expertise, as proposed in:
A version of this model for classification tasks is available here.
Sample multiple-annotator data using the MovieReviews dataset is provided here. More datasets are available here.
Copyright (c) 2016 Filipe Rodrigues
This program is free software. You can redistribute it and/or modify it under the terms of the GNU General Public License, version 3, as published by the Free Software Foundation.
This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details.
Appropriate reference to this software should be made when describing research in which it played a substantive role, so that it may be replicated and verified by others.
Compiling
Type "make" in a shell.
Please note that this code requires the Gnu Scientific Library, http://www.gnu.org/software/gsl/
Estimation
Usage:
./maslda est [data] [answers] [settings] [alpha] [tau] [k] [random/seeded/model_path] [seed] [directory]
Data format:
- [data] is a file where each line is of the form: [M] [term_1]:[count] [term_2]:[count] ... [term_N]:[count], where [M] is the number of unique terms in the document, and the [count] associated with each term is how many times that term appeared in the document.
- [answers] is a file where each line contains the target/response variable of the different annotators (separated by a white space) for [data]. Each column therefore corresponds to all the answers of an annotator.
Example:
./maslda est ../MovieReviews/data_train_amt.txt ../MovieReviews/answers.txt settings.txt 1 0.1 20 random 1 output
Inference
Usage:
./maslda inf [data] [label] [settings] [model] [directory]
Data format:
- [label] is a file where each line is the corresponding true target/response variable for [data].
Example:
./maslda inf ../MovieReviews/data_test.txt ../MovieReviews/labels_test.txt settings.txt output/final.model output
Settings
The settings file specifies the following parameters:
- "L2 penalty" controls the strength of the L2 regularization.
- "labels train file" is a file with the true target variables for the training documents. If a valid file is provided, it will be use to compute and report error statistics during the model estimation.
- "annotators quality file" is a file with the true biases and variances of the multiple annotators. If a valid file is provided, it will be use to compute and report error statistics during the model estimation.
- "lambda smoother" defines the values of the laplace smoothers used when estimating pi and lambda respectively.