Exemplo n.º 1
0
  /**
   * Select the method to create.
   *
   * @param dataset The dataset.
   * @param methodType The method type.
   */
  public void selectMethod(VersatileMLDataSet dataset, String methodType) {
    if (!this.methodConfigurations.containsKey(methodType)) {
      throw new EncogError("Don't know how to autoconfig method: " + methodType);
    }

    this.config = this.methodConfigurations.get(methodType);
    this.methodType = methodType;
    this.methodArgs = this.config.suggestModelArchitecture(dataset);
    dataset
        .getNormHelper()
        .setStrategy(this.config.suggestNormalizationStrategy(dataset, methodArgs));
  }
Exemplo n.º 2
0
 /**
  * Create the selected method.
  *
  * @return The created method.
  */
 public MLMethod createMethod() {
   if (this.methodType == null) {
     throw new EncogError(
         "Please call selectMethod first to choose what type of method you wish to use.");
   }
   MLMethodFactory methodFactory = new MLMethodFactory();
   MLMethod method =
       methodFactory.create(
           methodType,
           methodArgs,
           dataset.getNormHelper().calculateNormalizedInputCount(),
           this.config.determineOutputCount(dataset));
   return method;
 }