model_manager.py 文件源码

def build(self, environment, ml_definitions):
        self._messages.clear()

        # read received definitions and configure these
        field_manager = FieldManager.read_definitions(ml_definitions)
        field_manager.init(environment)

        # load dataset and evaluate
        dataset = DataSet.load_dataset(environment, field_manager=field_manager)
        self._merge_and_check_messages(dataset.evaluate())

        # build the feature from field and dataset
        f_builder = FeatureBuilder(field_manager)
        f_builder.build(dataset)
        self._merge_and_check_messages(f_builder.evaluate())

        # adjust the dataset to the feature
        adjusted = f_builder.field_manager.adjust(dataset)

        # make & train the model
        m_builder = ModelBuilder(f_builder.field_manager)
        m_builder.build(adjusted)
        self._merge_and_check_messages(m_builder.evaluate())

        self.field_manager = f_builder.field_manager
        self.model = m_builder.model
        self.model_score = m_builder.model_score