from pcfg import PCFG
from pcfg_logprob import LogProbPCFG

class PCFG_Predictor(nn.Module):
    def __init__(self, 
        feature_extractor,
        template_cfg):
        """
        feature_extractor: a neural network module 
        taking a list of tasks (each task is a list of input-outputs) 
        and returning a tensor of shape 
        [len(list_of_tasks), feature_extractor.output_dimensionality]

        template_cfg: a cfg giving the structure that will be output 