model.py 文件源码

def __init__(self,
                 mode='sequence',
                 random_state=1234):
        """Constructor.

        Params
        ------
        mode : str
            Values: 'sequence', 'rnafold' or 'rnaplfold'.

        n_jobs : int (default : -1)
            Number of jobs.

        random_state : int (default : 1234)
            Seed for random number generator.
        """
        self.mode = mode

        self.max_dist = None
        self.preprocessor_args = dict()
        self.vectorizer_args = dict()
        self.regressor_args = dict()
        self.smoothing_args = dict()

        if mode == 'sequence':
            self.preprocessor = seq.sequence_preprocessor
            self.vote_aggregator = seq.vote_aggregator
        elif mode == 'rnafold' or mode == 'rnaplfold':
            if mode == 'rnafold':
                self.preprocessor = graph.rnafold_preprocessor
            else:
                self.preprocessor = graph.rnaplfold_preprocessor
            self.vote_aggregator = graph.vote_aggregator
        else:
            raise Exception("Unrecognized mode: %s" % mode)
            exit(1)

        self.regressor = SGDRegressor(shuffle=True,
                                      random_state=random_state)

        # status variables
        self.is_optimized = False
        self.is_fitted = False