Spaces:
Sleeping
Sleeping
File size: 575 Bytes
f4af40d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
from app import app
from flask import request, jsonify
@app.route('/api/cluster', methods=['POST'])
def cluster():
data = request.get_json()
embeddings = data['embeddings']
from sklearn.cluster import DBSCAN
import numpy as np
try:
dbscan = DBSCAN(eps=0.4, min_samples=2, metric='cosine', n_jobs=-1)
embeddings_array = np.array(embeddings)
labels = dbscan.fit_predict(embeddings_array)
labels = labels.tolist()
return jsonify({'labels': labels})
except Exception as e:
return jsonify({'error': str(e)})
|