Dataford
Interview QuestionsInterview GuidesExperiencesMock InterviewsPricing
Get started

Implement an ML Algorithm in Python

HardPython00:00
Practice interviewer
In session
5 left
00:00

Your question is Implement an ML Algorithm in Python. Start with the requirements on the right.

Run and submit as often as you like. When you're ready, talk me through your approach or go straight to the code.

You need to log in / sign up to run or submit.

Problem

Mastech Digital's delivery analytics pipeline needs a dependency-free implementation of K-means clustering for grouping multidimensional service metrics. Implement the clustering algorithm without using machine learning libraries.

Given n points in d-dimensional space, divide them into exactly k clusters. Use deterministic farthest-point initialization: choose the first input point as the first centroid, then repeatedly choose the point whose distance to its nearest selected centroid is greatest. Break all distance ties by choosing the lowest input index.

Repeat these steps until convergence or max_iter iterations:

  1. Assign each point to its nearest centroid, breaking ties with the lowest centroid index.
  2. Replace each centroid with the coordinate-wise mean of its assigned points.
  3. If a cluster is empty, place its centroid at the point with the greatest distance to its assigned centroid, using the lowest index to break ties.

Return [labels, centroids], where labels[i] is the cluster index for points[i], and centroids is a list of k coordinate lists. Stop when every centroid moves by at most tol in Euclidean distance.

Constraints

  • 1 <= len(points) <= 10^4
  • 1 <= k <= len(points)
  • 1 <= len(points[i]) <= 20
  • Coordinates are integers in [-10^6, 10^6]
  • 1 <= max_iter <= 500
  • 0 <= tol <= 10^6
  • There are at least k distinct points

Function Signature

def k_means(points, k, max_iter, tol):
Your solutionPython 3
You need to log in / sign up to run or submit.
Run your code to see test output