()
| 23 | |
| 24 | |
| 25 | def main(): |
| 26 | X, Y = get_data(10000) |
| 27 | print("Number of data points:", len(Y)) |
| 28 | |
| 29 | model = GaussianMixture(n_components=10) |
| 30 | model.fit(X) |
| 31 | M = model.means_ |
| 32 | R = model.predict_proba(X) |
| 33 | |
| 34 | print("Purity:", purity(Y, R)) # max is 1, higher is better |
| 35 | print("DBI:", DBI(X, M, R)) # lower is better |
| 36 | |
| 37 | |
| 38 | if __name__ == "__main__": |
no test coverage detected