40 lines
1.1 KiB
Python
40 lines
1.1 KiB
Python
OPENAI_MODEL = "gpt-4o-mini"
|
|
CLAUDE_MODEL = "claude-3-5-haiku-20241022"
|
|
|
|
OUTPUT_MAX_TOKEN = 2000
|
|
|
|
PYTHON_CODE = '''
|
|
import math
|
|
|
|
def pairwise_distance(points_a, points_b):
|
|
"""
|
|
Compute the pairwise Euclidean distance between two sets of 3D points.
|
|
|
|
Args:
|
|
points_a: list of (x, y, z)
|
|
points_b: list of (x, y, z)
|
|
Returns:
|
|
A 2D list of shape (len(points_a), len(points_b)) representing distances
|
|
"""
|
|
distances = []
|
|
for i in range(len(points_a)):
|
|
row = []
|
|
for j in range(len(points_b)):
|
|
dx = points_a[i][0] - points_b[j][0]
|
|
dy = points_a[i][1] - points_b[j][1]
|
|
dz = points_a[i][2] - points_b[j][2]
|
|
d = math.sqrt(dx * dx + dy * dy + dz * dz)
|
|
row.append(d)
|
|
distances.append(row)
|
|
return distances
|
|
|
|
|
|
# Example usage
|
|
if __name__ == "__main__":
|
|
import random
|
|
points_a = [(random.random(), random.random(), random.random()) for _ in range(100)]
|
|
points_b = [(random.random(), random.random(), random.random()) for _ in range(100)]
|
|
dists = pairwise_distance(points_a, points_b)
|
|
print(f"Distance[0][0] = {dists[0][0]:.4f}")
|
|
'''
|