AutoTVM tree RNN model

merrymercy · December 11, 2018, 3:57am

Here is the related code. It is on my branch and not upstreamed.
You can find related tree-structure extraction code.

merrymercy/tvm/blob/autotvm-exp/python/tvm/autotvm/tuner/treegru_tuner.py

"""Tuner that uses xgboost as cost model"""

from .model_based_tuner import ModelBasedTuner, ModelOptimizer
from .treernn_cost_model import TreeRNNCostModel
from .sa_model_optimizer import SimulatedAnnealingOptimizer


class TreeGRUTuner(ModelBasedTuner):
    """Tuner that uses xgboost as cost model

    Parameters
    ----------
    task: Task
        The tuning task
    plan_size: int
        The size of a plan. After `plan_size` trials, the tuner will refit a new cost model
        and do planing for the next `plan_size` trials.
    feature_type: str, optional

    loss_type: str

This file has been truncated. show original