main.py

from dataviz import plot_data, plot_labeled_data
from dataset import get_datasets
from dbscan import DBSCAN, get_kdist_data


def main():
    datasets = get_datasets()

    min_points = 5
    eps = [20, 17, 11, 4]

    for i, dataset in enumerate(datasets):
        # Plot kdist plot to determine EPS param
        kdist_data = get_kdist_data(dataset, min_points)
        plot_data(kdist_data)

        # Get dbscan object
        dbscan = DBSCAN(min_points, eps[i])

        labels = dbscan.fit(dataset)

        print_labels(labels)

        plot_labeled_data(dataset, labels)


def print_labels(labels):
    print()
    for i, label in enumerate(labels):
        if i % 12 == 0:
            print()
        print('{}, '.format(label), end='')
    print()


if __name__ == '__main__':
    main()