ClassicComputerScienceProblemsInPython/Chapter6/mj.py at master · davecom/ClassicComputerScienceProblemsInPython · GitHub

43 lines (38 loc) · 1.9 KB

# mj.py

# From Classic Computer Science Problems in Python Chapter 6

# Licensed under the Apache License, Version 2.0 (the "License");

# you may not use this file except in compliance with the License.

# You may obtain a copy of the License at

# http://www.apache.org/licenses/LICENSE-2.0

# Unless required by applicable law or agreed to in writing, software

# distributed under the License is distributed on an "AS IS" BASIS,

# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

# See the License for the specific language governing permissions and

# limitations under the License.

from __future__ import annotations

from typing import List

from data_point import DataPoint

from kmeans import KMeans

class Album(DataPoint):

def __init__(self, name: str, year: int, length: float, tracks: float) -> None:

super().__init__([length, tracks])

self.name = name

self.year = year

self.length = length

self.tracks = tracks

def __repr__(self) -> str:

return f"{self.name}, {self.year}"

if __name__ == "__main__":

albums: List[Album] = [Album("Got to Be There", 1972, 35.45, 10), Album("Ben", 1972, 31.31, 10),

Album("Music & Me", 1973, 32.09, 10), Album("Forever, Michael", 1975, 33.36, 10),

Album("Off the Wall", 1979, 42.28, 10), Album("Thriller", 1982, 42.19, 9),

Album("Bad", 1987, 48.16, 10), Album("Dangerous", 1991, 77.03, 14),

Album("HIStory: Past, Present and Future, Book I", 1995, 148.58, 30), Album("Invincible", 2001, 77.05, 16)]

kmeans: KMeans[Album] = KMeans(2, albums)

clusters: List[KMeans.Cluster] = kmeans.run()

for index, cluster in enumerate(clusters):

print(f"Cluster {index} Avg Length {cluster.centroid.dimensions[0]} Avg Tracks {cluster.centroid.dimensions[1]}: {cluster.points}\n")

Expand file tree