Skip to main content

An intent classification library for AI Models.

Project description

Intentify

A python library for training intent classification models.

Setting Up A Basic Model

from intentify import Intentify

file_path = 'intents.json'
name = 'test_model'

Model = Intentify(file_path, model_name=name)
Model.train_model()
Model.save_model()

done = False

while not done:
    message = input("You: ")
    if message == "STOP":
        done = True
    else:
        response = Model.request(message)
        print(response)

Binding Functions To Requests

from intentify import Intentify

def function_for_greetings():
    print("Greetings Intent Trigger!")
    # Some action you want to take

def function_for_stocks():
    print("Stocks Intent Trigger!")
    # Some action you want to take

mappings = {'greeting' : function_for_greetings, 'stocks' : function_for_stocks}
file_path = 'intents.json'
name = 'test_model'

Model = Intentify(file_path, intent_methods=mappings ,model_name=name)
Model.train_model()
Model.save_model()

done = False

while not done:
    message = input("Enter a message: ")
    if message == "STOP":
        done = True
    else:
        Model.request(message)

Once you have saved the model you can simply call "Model.load_model()" instead of ".train_model()" and ".save_model()"

Sample intents.json File

{"intents": [
  {"tag": "greeting",
    "patterns": ["Hi", "How are you", "Is anyone there?", "Hello", "Good day", "Whats up", "Hey", "greetings"],
    "responses": ["Hello!", "Good to see you again!", "Hi there, how can I help?"],
    "context_set": ""
  },
  {"tag": "goodbye",
    "patterns": ["cya", "See you later", "Goodbye", "I am Leaving", "Have a Good day", "bye", "cao", "see ya"],
    "responses": ["Sad to see you go :(", "Talk to you later", "Goodbye!"],
    "context_set": ""
  },
  {"tag": "stocks",
    "patterns": ["what stocks do I own?", "how are my shares?", "what companies am I investing in?", "what am I doing in the markets?"],
    "responses": ["You own the following shares: ABBV, AAPL, FB, NVDA and an ETF of the S&P 500 Index!"],
    "context_set": ""
  }
]
}

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

intentify-0.0.3.tar.gz (4.1 kB view hashes)

Uploaded Source

Built Distribution

intentify-0.0.3-py3-none-any.whl (4.2 kB view hashes)

Uploaded Python 3

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page