hivemall-issues mailing list archives

Site index · List index
Message view « Date » · « Thread »
Top « Date » · « Thread »
From myui <...@git.apache.org>
Subject [GitHub] incubator-hivemall pull request #14: [WIP] Separate optimizer implementation...
Date Thu, 26 Jan 2017 07:12:46 GMT
Github user myui commented on a diff in the pull request:

    https://github.com/apache/incubator-hivemall/pull/14#discussion_r97715666
  
    --- Diff: core/src/main/java/hivemall/classifier/GeneralClassifierUDTF.java ---
    @@ -0,0 +1,122 @@
    +/*
    + * Hivemall: Hive scalable Machine Learning Library
    + *
    + * Copyright (C) 2015 Makoto YUI
    + * Copyright (C) 2013-2015 National Institute of Advanced Industrial Science and Technology
(AIST)
    + *
    + * Licensed under the Apache License, Version 2.0 (the "License");
    + * you may not use this file except in compliance with the License.
    + * You may obtain a copy of the License at
    + *
    + *         http://www.apache.org/licenses/LICENSE-2.0
    + *
    + * Unless required by applicable law or agreed to in writing, software
    + * distributed under the License is distributed on an "AS IS" BASIS,
    + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
    + * See the License for the specific language governing permissions and
    + * limitations under the License.
    + */
    +package hivemall.classifier;
    +
    +import java.util.HashMap;
    +import java.util.Map;
    +import javax.annotation.Nonnull;
    +
    +import org.apache.commons.cli.CommandLine;
    +import org.apache.commons.cli.Options;
    +import org.apache.hadoop.hive.ql.exec.UDFArgumentException;
    +import org.apache.hadoop.hive.serde2.objectinspector.ObjectInspector;
    +import org.apache.hadoop.hive.serde2.objectinspector.StructObjectInspector;
    +
    +import hivemall.optimizer.LossFunctions;
    +import hivemall.model.FeatureValue;
    +
    +/**
    + * A general classifier class with replaceable optimization functions.
    + */
    +public class GeneralClassifierUDTF extends BinaryOnlineClassifierUDTF {
    +
    +    protected final Map<String, String> optimizerOptions;
    +
    +    public GeneralClassifierUDTF() {
    +        super(true); // This enables new model interfaces
    +        this.optimizerOptions = new HashMap<String, String>();
    +        // Set default values
    +        optimizerOptions.put("optimizer", "adagrad");
    +        optimizerOptions.put("eta", "fixed");
    +        optimizerOptions.put("eta0", "1.0");
    +        optimizerOptions.put("regularization", "RDA");
    +        optimizerOptions.put("lambda", "1e-6");
    +        optimizerOptions.put("scale", "100.0");
    +        optimizerOptions.put("lambda", "1.0");
    +    }
    +
    +    @Override
    +    public StructObjectInspector initialize(ObjectInspector[] argOIs) throws UDFArgumentException
{
    +        if(argOIs.length != 2 && argOIs.length != 3) {
    +            throw new UDFArgumentException(
    +                    this.getClass().getSimpleName()
    +                  + " takes 2 or 3 arguments: List<Text|Int|BitInt> features, int
label "
    +                  + "[, constant string options]");
    +        }
    +        return super.initialize(argOIs);
    +    }
    +
    +    @Override
    +    protected Options getOptions() {
    +        Options opts = super.getOptions();
    +        opts.addOption("optimizer", "opt", true, "Optimizer to update weights [default:
adagrad+rda]");
    +        opts.addOption("eta", "eta0", true, "Initial learning rate [default 1.0]");
    +        opts.addOption("lambda", true, "Lambda value of RDA [default: 1e-6f]");
    +        opts.addOption("scale", true, "Scaling factor for cumulative weights [100.0]");
    +        opts.addOption("regularization", "reg", true, "Regularization type [default not-defined]");
    +        opts.addOption("lambda", true, "Regularization term on weights [default 1.0]");
    +        return opts;
    +    }
    +
    +    @Override
    +    protected CommandLine processOptions(ObjectInspector[] argOIs) throws UDFArgumentException
{
    +        final CommandLine cl = super.processOptions(argOIs);
    +        assert(cl != null);
    +        if(cl != null) {
    +            for(final String arg : cl.getArgs()) {
    +                optimizerOptions.put(arg, cl.getOptionValue(arg));
    +            }
    +        }
    +        return cl;
    +    }
    +
    +    @Override
    +    protected Map<String, String> getOptimzierOptions() {
    --- End diff --
    
    removed `getOptimzierOptions`


---
If your project is set up for it, you can reply to this email and have your
reply appear on GitHub as well. If your project does not have this feature
enabled and wishes so, or if the feature is enabled but not working, please
contact infrastructure at infrastructure@apache.org or file a JIRA ticket
with INFRA.
---

Mime
View raw message