US 12,332,926 B2
Unit prediction for custom feature engineering
Lei Liu, Sunnyvale, CA (US); Yinlin Deng, Sunnyvale, CA (US); Mehdi Bahrami, Santa Clara, CA (US); and Mukul R. Prasad, San Jose, CA (US)
Assigned to Fujitsu Limited, Kawasaki (JP)
Filed by Fujitsu Limited, Kawasaki (JP)
Filed on Nov. 16, 2022, as Appl. No. 17/988,651.
Prior Publication US 2024/0160646 A1, May 16, 2024
Int. Cl. G06F 16/3329 (2025.01); G06F 16/355 (2025.01)
CPC G06F 16/3329 (2019.01) [G06F 16/355 (2019.01)] 17 Claims
OG exemplary drawing
 
1. A method, comprising:
obtaining a dataset having one or more columns, each of the one or more columns includes a title and at least one value;
obtaining a task description corresponding to a machine learning model to be trained using the dataset;
extracting, for each of the one or more columns, the title and a sample value lacking a corresponding unit of measurement from the at least one value;
synthesizing a question to elicit, from a language model, an answer indicating a predicted unit of measurement predicting the corresponding unit of measurement associated with the sample value from each of the one or more columns, the question synthesized based on:
the title for each of the one or more columns;
the sample value for each of the one or more columns; and
the task description corresponding to the machine learning model to be trained using the dataset;
sending the question to the language model to obtain an answer; and
generating from the answer to the question, the predicted unit of measurement for the at least one value in each of the one or more columns;
adding the predicted unit of measurement to the at least one value in each of the one or more columns in the dataset; and
training the machine learning model using the dataset including the added predicted unit of measurement.