Modelpublic

markrodrigo/Qwen-3.5-2B-Spatial-SQL-1.1

sourceHugging Faceapache-2.0updated 20d agoView on Hugging Face

0likes346downloads

Model Card

Note

This is experimental. Models are for prototyping this general approach of text to spatial SQL.

Model Information

This model, Qwen-3.5-2B-Spatial-SQL-1.1, is a 2B, narrow use case, text to spatial SQL, lightly fine-tuned model. In general, its primary use case is the Natural Language command adaptation of particular geographic spatial functions as normally defined in pure SQL. Data input should be a combination of an English prefix in the form of a question, and a coordinate prompt injection, likely from an active mapping system application coordinate list. Output is PostGIS spatial SQL.

There are five primary geographic functions released in version 1.1.

Model developer: Mark Rodrigo

Github: https://github.com/mprodrigo/spatialsql

Model Architecture: Supervised Fine Tuning (SFT)

Model Input / Output Overview:

Input: Text plus coordinate prompt injection. Output: PostGIS spatial SQL NOTE: Inputs and outputs are in meters and or geographic decimal degrees WGS 84 coordinates.

Function	Question Input	Geo Input	SQL Execution Output
Area	Area question	Polygon	Number - Area sq meters
Centroid	Center question	Polygon	Point
Buffer	Buffer distance	Point	Polygon
Length	Length question	Line	Number - Length in meters

Function	Question Input	Geo Input 1	Geo Input 2	SQL Execution Output
Distance	Distance question	Point, Line, or Polygon	Point, Line, or Polygon	Number - Distance between geometries in meters

Example Prompting

pipe = pipeline(task="text-generation", model=model, tokenizer=tokenizer, maxnewtokens=512, temperature=0.4, returnfulltext=False,)

Build the messages messages = [ {"role": "system", "content": "Write a PostGIS SQL statement for the following."}, {"role": "user", "content": "What is the thousand meter buffer for the following point? : 'Point(-3.78621945 40.4463195)'"}, ]

Apply ChatML Template prompt = tokenizer.applychattemplate( messages, tokenize=False, addgenerationprompt=True, enable_thinking=False, )

Examples

AREA

Message: What is the area for the polygon? : 'Polygon ((-3.7515154 40.3855551, -3.7514972 40.3856581, -3.7507005 40.3855767, -3.7507167 40.3854722, -3.7515154 40.3855551))'

Response: SELECT ST_Area(geog) As area FROM (select 'Polygon ((-3.7515154 40.3855551, -3.7514972 40.3856581, -3.7507005 40.3855767, -3.7507167 40.3854722, -3.7515154 40.3855551))' :: geography geog) subquery;

CENTROID

Message: What is the centroid for the polygon? : 'Polygon ((-3.6934636 40.4808785, -3.6933352 40.4811486, -3.6930125 40.4810598, -3.693141 40.4807897, -3.6934636 40.4808785))'

Response: SELECT STAsText(STCentroid(geog)) As centroid FROM (select 'Polygon ((-3.6934636 40.4808785, -3.6933352 40.4811486, -3.6930125 40.4810598, -3.693141 40.4807897, -3.6934636 40.4808785))' :: geography geog) subquery;

BUFFER

Message: What is the thousand meter buffer for the following point? : 'Point(-8.7522658 41.3862664)'

Response: SELECT STAsText(STBuffer(geog, 1000)) as buffer FROM (select 'Point(-8.7522658 41.3862664)' :: geography geog) subquery;

LENGTH

Message: How long is the line? : 'LINESTRING (-3.6976693 40.4263178, -3.6986082 40.4258729)'

Response: SELECT ST_Length(geog) As length FROM (select 'LINESTRING (-3.6976693 40.4263178, -3.6986082 40.4258729)' :: geography geog) subquery; DISTANCE

Message: How far apart is the point and line? : 'Point(-109.87549823 38.60574249)' 'LineString(-109.24324628 38.76349931, -109.4821773 38.6875815)'

Response: SELECT ST_Distance(gg1, gg2) As distance FROM (select 'Point(-109.87549823 38.60574249)' :: geography gg1, 'LineString(-109.24324628 38.76349931, -109.4821773 38.6875815)' :: geography gg2) subquery;

A Few Known Question Variation Examples

AREA What is the area for the geometry? What is the area for this polygon? CENTROID What is the centroid for the geometry? What is the center point of the polygon? BUFFER What is the 100 meter buffer for the following point? Buffer the following point a thousand meters. What is the 1000 meter buffer for the following point? LENGTH What is the length of the line? How long is this line? DISTANCE The distance between the polygon and the line is? What is the distance between the points? How far apart are the two lines?

llama.cpp / Hyperparameter Recommendations For Inference

max context ~ 262,000 top k ~ 100 temp ~ .4-.5 or lower

Agent Considerations

Agents are being considered as a separate project. Agents would mostly be related to pulling the coordinates from a mapping UI, and executing the SQL from responses against a PostGIS database.

Further Reference - link this

https://postgis.net/docs/PostGISSpecialFunctionsIndex.html#PostGISGeographyFunctions

Evaluation data

More information needed

Training data

Custom synthetic

Training hyperparameters

The following hyperparameters were used during training:

—learning_rate: 1e-4
—distributed_type: multi-GPU
—num_devices: 2
—optimizer: Adam 8bit
—lrschedulertype: linear

Training results

Training Loss	Step	Validation Loss
1.0856	10	0.8841
0.7787	20	0.7227
0.7160	30	0.6690
0.6631	40	0.6226
0.5834	50	0.5627
0.5230	60	0.5085
0.4599	70	0.4838

Framework versions

—transformers 5.14.1
—torch 2.13.0
—peft 0.20.0
—bitsandbytes 0.50.0
—datasets 5.0.1
—tokenizers 0.23.1