A method and system for data transformation based on machine learning is disclosed. The method includes generating a matrix for a plurality of input vectors based on a machine learning model. The method further includes comparing for each of the plurality of input vectors, the intent value in the matrix with a predefined intent threshold, wherein, for an intent value below the predefined intent threshold, an associated function is unavailable. The method further includes determining a first set of vectors from the plurality of input vectors based on the comparing, wherein for each input vector in the first set, the associated intent value is below the predefined intent threshold. The method further includes mapping, by the data transformation device, each input vector in the first set with an intent value above the predefined intent threshold and an associated function.