gender_plots.sh

#/bin/bash!

THIS=`basename $0`
HERE=$(pwd)
NOW=`date +%Y%m%D-%H%M%S`
SRC_DIR=../src
WORD2VEC_DIR=../../word2vec
BIN_DIR=$WORD2VEC_DIR/bin


function usage() {
    printf "\nUsage: $THIS [-t PATH_TO_TXT_FILE] [-g PATH_TO_BIN]\n"
    printf "\nARGUMENTS\n"
    printf "\n    -g, --train        Train the model on a specific data set"
    printf "\n    -p, --plot         Generate a plot from a vector reading bin\n"
    printf "\n    -h, --help        Train the model on a specific data set"
    echo

    return 0
}


function dafault_action {
  # When no arguments provided this action is called
  # Build the binary
  if [[ ! -d "bin" ]]; then
    mkdir bin
  fi

  echo "Bulding genplot binary"
  gcc src/gendered_word_plotting.c -o bin/genplot -lm
  # -lm require to properly ling <math.h> function on Ubuntu
  # See here for more info https://stackoverflow.com/questions/5248919/undefined-reference-to-sqrt-or-other-mathematical-functions
}


if [[ $# = 0 ]]; then
    dafault_action
    exit 0
fi

# Parse Opts
#while the number of arguments is greater than

while [[ $# -gt 0 ]]; do
    #check signal against the following options
    case $1 in
        --train|-t)
            if [ -z "$2" ]; then
                echo "Must specify a text file to train your model. Default model has already been trained on text8"
                shift 1
            else
                echo "Training on data set $2"
                TEXT_DATA=$2
                VECTOR_DATA="${TEXT_DATA%%.txt}"-vector.bin
                TASK=1
                shift 2
            fi
            ;;
        --plot|-p)
            GENMAP=1
            if [ -z "$2" ]; then
                echo "Must specify the vector bin that you are referencing for your plot"
                echo "Plotting using default text8 vector bin"
                VECTOR_DATA=../vectorbins/text8-vector.bin
                shift 1
            else
                VECTOR_DATA=$2
                shift 2
            fi
            ;;
        --help|-h)
            # Print help and exit
            usage
            shift 1
            ;;
        *)
            usage
            shift 1
            ;;
    esac
done



if [ -z "$TASK" ]; then
    echo
else
    #check if word2vec model already exists in the folder above. clone it if it does not exist
    if [ -d $WORD2VEC_DIR ]; then
        echo "Word2vec already cloned. Cool"
    else
        git clone https://github.com/dav/word2vec.git $WORD2VEC_DIR
    fi
    pushd $WORD2VEC_DIR/src && make; popd
    time $BIN_DIR/word2vec -train $TEXT_DATA -output $VECTOR_DATA -cbow 0 -size 200 -window 5 -negative 0 -hs 1 -sample 1e-3 -threads 12 -binary 1
fi


if [ -z "$GENMAP"  ]; then
    echo
else
    pushd $SRC_DIR
    #project the points and generate a word plot
    gcc gendered_word_plotting.c
    ./a.out $VECTOR_DATA
    python word_plots.py
    popd

fi