data_analysis.py 文件源码

def detect_outlier(data_train, measurement):
    """
    Detect whether the input measurement is outlier or not.
    :param data_train: data for training the one class SVM model
    :param measurement: one row from the chill_untested.csv
    :return: predicted label for input measurement
    """
    classifier = svm.OneClassSVM(kernel="rbf", nu=0.005, gamma=0.00001)
    classifier.fit(data_train)
    label = classifier.predict(measurement)[0]
    return label