Example #1
0
 private RoleVar createRoleVar(
     int parent, int child, IntSet knownPreds, List<String> roleStateNames) {
   RoleVar roleVar;
   String roleVarName = "Role_" + parent + "_" + child;
   if (!prm.makeUnknownPredRolesLatent || knownPreds.contains((Integer) parent)) {
     roleVar =
         new RoleVar(
             VarType.PREDICTED, roleStateNames.size(), roleVarName, roleStateNames, parent, child);
   } else {
     roleVar =
         new RoleVar(
             VarType.LATENT, roleStateNames.size(), roleVarName, roleStateNames, parent, child);
   }
   return roleVar;
 }
Example #2
0
  /** Adds factors and variables to the given factor graph. */
  public void build(
      List<String> words,
      List<String> lemmas,
      IntSet knownPreds,
      List<String> roleStateNames,
      Map<String, List<String>> psMap,
      ObsFeatureExtractor obsFe,
      ObsFeatureConjoiner ofc,
      FactorGraph fg) {
    // Check for null arguments.
    if (prm.roleStructure == RoleStructure.PREDS_GIVEN && knownPreds == null) {
      throw new IllegalArgumentException("knownPreds must be non-null");
    }
    if (prm.predictSense && (lemmas == null || psMap == null)) {
      throw new IllegalArgumentException("lemmas and psMap must be non-null");
    }
    if (prm.roleStructure == RoleStructure.PREDS_GIVEN && prm.predictPredPos) {
      throw new IllegalStateException(
          "PREDS_GIVEN assumes that the predicate positions are always observed.");
    }

    this.n = words.size();

    // Create the Role variables.
    roleVars = new RoleVar[n][n];
    if (prm.roleStructure == RoleStructure.PREDS_GIVEN) {
      // CoNLL-friendly model; preds given
      IntIter iter = knownPreds.iterator();
      while (iter.hasNext()) {
        int i = iter.next();
        for (int j = 0; j < n; j++) {
          if (i == j && !prm.allowPredArgSelfLoops) {
            continue;
          }
          roleVars[i][j] = createRoleVar(i, j, knownPreds, roleStateNames);
        }
      }
    } else if (prm.roleStructure == RoleStructure.ALL_PAIRS) {
      // n**2 model
      for (int i = 0; i < n; i++) {
        for (int j = 0; j < n; j++) {
          if (i == j && !prm.allowPredArgSelfLoops) {
            continue;
          }
          roleVars[i][j] = createRoleVar(i, j, knownPreds, roleStateNames);
        }
      }
    } else {
      throw new IllegalArgumentException("Unsupported model structure: " + prm.roleStructure);
    }

    // Create the Sense variables.
    senseVars = new SenseVar[n];
    for (int i = 0; i < n; i++) {
      // Only look at the knownPreds if the predicate positions are given.
      if (prm.roleStructure == RoleStructure.PREDS_GIVEN && !knownPreds.contains(i)) {
        // Skip non-predicate positions.
        continue;
      }
      if (prm.roleStructure == RoleStructure.ALL_PAIRS && prm.predictSense && prm.predictPredPos) {
        // Sense and position.
        List<String> senseStateNames = psMap.get(lemmas.get(i));
        if (senseStateNames == null) {
          senseStateNames = CorpusStatistics.SENSES_FOR_UNK_PRED;
        }
        // Include the state of "no predicate".
        senseStateNames = Lists.cons("_", senseStateNames);
        senseVars[i] = createSenseVar(i, senseStateNames);
      } else if (prm.predictSense) {
        // Sense without positions.
        List<String> senseStateNames = psMap.get(lemmas.get(i));
        if (senseStateNames == null) {
          senseStateNames = CorpusStatistics.SENSES_FOR_UNK_PRED;
        }
        senseVars[i] = createSenseVar(i, senseStateNames);
      } else if (prm.predictPredPos) {
        // Positions without sense.
        senseVars[i] = createSenseVar(i, CorpusStatistics.PRED_POSITION_STATE_NAMES);
      }
    }

    // Add the factors.
    for (int i = -1; i < n; i++) {
      // Get the lemma or UNK if we don't know it.
      String lemmaForTk = null;
      if (i >= 0) {
        if (prm.predictSense && psMap.get(lemmas.get(i)) != null) {
          // The template key must include the lemma appended, so that
          // there is a unique set of model parameters for each predicate.
          lemmaForTk = lemmas.get(i);
        } else {
          // If we've never seen this predicate, just give it to the (untrained) unknown classifier.
          lemmaForTk = CorpusStatistics.UNKNOWN_SENSE;
        }
      }
      // Add the unary factors for the sense variables.
      if (i >= 0 && senseVars[i] != null && senseVars[i].getType() != VarType.OBSERVED) {
        String templateKey = SrlFactorTemplate.SENSE_UNARY + "_" + lemmaForTk;
        fg.addFactor(
            new ObsFeTypedFactor(
                new VarSet(senseVars[i]), SrlFactorTemplate.SENSE_UNARY, templateKey, ofc, obsFe));
      }
      // Add the role factors.
      for (int j = 0; j < n; j++) {
        if (i != -1) {
          // Add unary factors on Roles.
          if (prm.unaryFactors && roleVars[i][j] != null) {
            fg.addFactor(
                new ObsFeTypedFactor(
                    new VarSet(roleVars[i][j]), SrlFactorTemplate.ROLE_UNARY, ofc, obsFe));
          }
          // Add binary factors between Role and Sense variables.
          if (prm.binarySenseRoleFactors && senseVars[i] != null && roleVars[i][j] != null) {
            String templateKey = SrlFactorTemplate.SENSE_ROLE_BINARY + "_" + lemmaForTk;
            fg.addFactor(
                new ObsFeTypedFactor(
                    new VarSet(senseVars[i], roleVars[i][j]),
                    SrlFactorTemplate.SENSE_ROLE_BINARY,
                    templateKey,
                    ofc,
                    obsFe));
          }
        }
      }
    }
  }