nirdizati_light.recommender_model.recommender_model 

 1from nirdizati_light.recommender_model.classifier_and_regressor.classifier_and_regressor import ClassifierAndRegressor
 2from nirdizati_light.recommender_model.common import RecommendationMethods
 3
 4
 5class RecommenderModel:
 6    def __init__(self,
 7                 CONF,
 8                 CONF_CLASSIFIER, classifier_train_df, classifier_validate_df,
 9                 CONF_REGRESSOR, regressor_train_df, regressor_validate_df):
10        self.model_type = CONF['recommender_model']
11        self.model = None
12        self.CONF_CLASSIFIER = CONF_CLASSIFIER
13        self.classifier_train_df = classifier_train_df
14        self.classifier_validate_df = classifier_validate_df
15        self.CONF_REGRESSOR = CONF_REGRESSOR
16        self.regressor_train_df = regressor_train_df
17        self.regressor_validate_df = regressor_validate_df
18        self._instantiate_model()
19
20    def retrieve_best_model(self, max_evaluations, target):
21        self.model.fit(max_evaluations, target)
22
23    def _instantiate_model(self):
24        if self.model_type is RecommendationMethods.CLASSIFICATION_AND_REGRESSION.value:
25            self.model = ClassifierAndRegressor(
26                self.model_type,
27                self.CONF_CLASSIFIER,
28                self.classifier_train_df,
29                self.classifier_validate_df,
30                self.CONF_REGRESSOR,
31                self.regressor_train_df,
32                self.regressor_validate_df)
33        else:
34            raise Exception('unsupported recommender')
class RecommenderModel: 
 6class RecommenderModel:
 7    def __init__(self,
 8                 CONF,
 9                 CONF_CLASSIFIER, classifier_train_df, classifier_validate_df,
10                 CONF_REGRESSOR, regressor_train_df, regressor_validate_df):
11        self.model_type = CONF['recommender_model']
12        self.model = None
13        self.CONF_CLASSIFIER = CONF_CLASSIFIER
14        self.classifier_train_df = classifier_train_df
15        self.classifier_validate_df = classifier_validate_df
16        self.CONF_REGRESSOR = CONF_REGRESSOR
17        self.regressor_train_df = regressor_train_df
18        self.regressor_validate_df = regressor_validate_df
19        self._instantiate_model()
20
21    def retrieve_best_model(self, max_evaluations, target):
22        self.model.fit(max_evaluations, target)
23
24    def _instantiate_model(self):
25        if self.model_type is RecommendationMethods.CLASSIFICATION_AND_REGRESSION.value:
26            self.model = ClassifierAndRegressor(
27                self.model_type,
28                self.CONF_CLASSIFIER,
29                self.classifier_train_df,
30                self.classifier_validate_df,
31                self.CONF_REGRESSOR,
32                self.regressor_train_df,
33                self.regressor_validate_df)
34        else:
35            raise Exception('unsupported recommender')
RecommenderModel( CONF, CONF_CLASSIFIER, classifier_train_df, classifier_validate_df, CONF_REGRESSOR, regressor_train_df, regressor_validate_df) 
 7    def __init__(self,
 8                 CONF,
 9                 CONF_CLASSIFIER, classifier_train_df, classifier_validate_df,
10                 CONF_REGRESSOR, regressor_train_df, regressor_validate_df):
11        self.model_type = CONF['recommender_model']
12        self.model = None
13        self.CONF_CLASSIFIER = CONF_CLASSIFIER
14        self.classifier_train_df = classifier_train_df
15        self.classifier_validate_df = classifier_validate_df
16        self.CONF_REGRESSOR = CONF_REGRESSOR
17        self.regressor_train_df = regressor_train_df
18        self.regressor_validate_df = regressor_validate_df
19        self._instantiate_model()
model_type
model
CONF_CLASSIFIER
classifier_train_df
classifier_validate_df
CONF_REGRESSOR
regressor_train_df
regressor_validate_df
def retrieve_best_model(self, max_evaluations, target): 
21    def retrieve_best_model(self, max_evaluations, target):
22        self.model.fit(max_evaluations, target)