gabrielhuang

import torch
from torch import nn
import torch.nn.functional as F
from torch.nn.modules import transformer

embedding_size = 16

vocab = "abcdefghijklmnopqrstuvwxyz .!?'\"\n$"  # $ is also EOS
vocab_size = len(vocab)

gabrielhuang

FROM grobid/grobid:0.7.2

RUN apt-get update && \
    apt-get install openjdk-8-jdk -y --no-install-recommends && \
    apt-get autoremove -y --purge && \
    apt-get clean -y && \
    rm -rf /var/lib/apt/lists/* /tmp/* /var/tmp/* && \
    rm -f /etc/legal /etc/motd

WORKDIR /

gabrielhuang

import cmudict  # for syllables
import syllables
import re

whitespace = re.compile(r'[\s,.?!/=();]+')
cmudict_cached = cmudict.dict()


def lookup_word(word_s):

gabrielhuang

import os, sys
sys.path.append(os.getcwd())

import time

import matplotlib
matplotlib.use('Agg')
import matplotlib.pyplot as plt
import numpy as np
import sklearn.datasets