Eurosmart commited on
Commit
4150fe4
·
1 Parent(s): bd8a530

change eps and minsamplkes

Browse files
Files changed (1) hide show
  1. api/routes/clusterRoute.py +19 -1
api/routes/clusterRoute.py CHANGED
@@ -1,13 +1,31 @@
1
  from app import app
2
  from flask import request, jsonify
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
3
  @app.route('/api/cluster', methods=['POST'])
4
  def cluster():
5
  data = request.get_json()
6
  embeddings = data['embeddings']
 
7
  from sklearn.cluster import DBSCAN
8
  import numpy as np
9
  try:
10
- dbscan = DBSCAN(eps=0.2, min_samples=5, metric='cosine', n_jobs=-1)
 
11
  embeddings_array = np.array(embeddings)
12
  labels = dbscan.fit_predict(embeddings_array)
13
  labels = labels.tolist()
 
1
  from app import app
2
  from flask import request, jsonify
3
+
4
+ CONFIG = {
5
+ "day": {
6
+ "eps": 0.2,
7
+ "min_samples": 5
8
+ },
9
+ "week": {
10
+ "eps": 0.1,
11
+ "min_samples": 7
12
+ },
13
+ "month": {
14
+ "eps": 0.1,
15
+ "min_samples": 7,
16
+ },
17
+ }
18
+
19
  @app.route('/api/cluster', methods=['POST'])
20
  def cluster():
21
  data = request.get_json()
22
  embeddings = data['embeddings']
23
+ duration = data['duration']
24
  from sklearn.cluster import DBSCAN
25
  import numpy as np
26
  try:
27
+
28
+ dbscan = DBSCAN(eps=CONFIG[duration]['eps'], min_samples=CONFIG[duration]['min_samples'], metric='cosine', n_jobs=-1)
29
  embeddings_array = np.array(embeddings)
30
  labels = dbscan.fit_predict(embeddings_array)
31
  labels = labels.tolist()