数理コンサルタントの備忘録

あなたの悩みを数理で解決する

python classの書き方例

分析が進むと、python classを記述する必要がある。 compute_KNN_featuresソースコードを参考にして、KNNのクラス概要をまとめた。

class NearestNeighborsFeats(BaseEstimator, ClassifierMixin):
    '''
        This class should implement KNN features extraction 
    '''
    # __init__関数は、インスタンス化を行う時に必ず最初に呼び出されるメソッド
    def __init__(self, n_jobs, k_list, metric, n_classes=None, n_neighbors=None, eps=1e-6):
        self.n_jobs = n_jobs # selfはインスタンス自身。self以降の引数を代入する
        self.k_list = k_list
        self.metric = metric
        self.n_classes_ = n_classes
        self.eps = eps

    def fit(self, X, y):
        '''
            Set's up the train set and self.NN object
        '''

    
    def predict(self, X):       
        '''
            Produces KNN features for every object of a dataset X
        '''


    def get_features_for_one(self, x):
        '''
            Computes KNN features for a single object `x`
        '''