Skip to content
Snippets Groups Projects
Commit ab0578c6 authored by Simonas's avatar Simonas
Browse files

chore: Encoder

parent 39266c64
No related branches found
No related tags found
No related merge requests found
...@@ -3,6 +3,7 @@ from typing import List ...@@ -3,6 +3,7 @@ from typing import List
import numpy as np import numpy as np
from matplotlib import pyplot as plt from matplotlib import pyplot as plt
from nltk.tokenize import word_tokenize from nltk.tokenize import word_tokenize
from semantic_router.encoders.base import BaseEncoder
from semantic_router.schema import DocumentSplit from semantic_router.schema import DocumentSplit
from semantic_router.splitters.base import BaseSplitter from semantic_router.splitters.base import BaseSplitter
...@@ -35,7 +36,7 @@ class RollingWindowSplitter(BaseSplitter): ...@@ -35,7 +36,7 @@ class RollingWindowSplitter(BaseSplitter):
def __init__( def __init__(
self, self,
encoder, encoder: BaseEncoder,
score_threshold=0.3, score_threshold=0.3,
window_size=5, window_size=5,
plot_splits=False, plot_splits=False,
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment