Improving the results and process of machine learning service in computer program development. A client's codebase is accessed. A set of features are extracted from the client's codebase. One or more features from the set of features are then selected. Thereafter, at least one of the selected features is sent to a machine learning service that uses the received feature(s) to build custom model(s) for the client's computer system.