TextMatch/textmatch/tools/clustering/dbscan.py at master · MachineLP/TextMatch · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
# -*- coding:utf-8 -*-
'''
-------------------------------------------------
   Description :  DBSCAN
   Author :       machinelp
   Date :         2020-06-15
-------------------------------------------------

'''

import numpy as np
from sklearn.cluster import DBSCAN


class DBSCANClustering(object):
    def __init__(self, eps=0.5, min_samples=5):
        self.db_scan = DBSCAN(eps=eps, min_samples=min_samples)

    def fit_(self, data_list):
        self.db_scan.fit(data_list)
        return self

    def predict_(self, data_list):
        label_list = self.db_scan.predict(data_list)
        return label_list

    def predict(self, data_list):
        label_list = self.db_scan.fit_predict(data_list)
        return label_list

    def save_model_(self, save_path):
        #保存模型
        joblib.dump(self.db_scan, save_path)