krishnaveni76 commited on
Commit
6afb80a
·
verified ·
1 Parent(s): c3c7748

updates entity

Browse files
anime_recommender/entity/artifact_entity.py CHANGED
@@ -1,21 +1,21 @@
1
- from dataclasses import dataclass
2
- from typing import Optional
3
-
4
- @dataclass
5
- class DataIngestionArtifact:
6
- feature_store_anime_file_path:str
7
- feature_store_userrating_file_path:str
8
-
9
- @dataclass
10
- class DataTransformationArtifact:
11
- merged_file_path:str
12
-
13
- @dataclass
14
- class CollaborativeModelArtifact:
15
- svd_file_path: Optional[str] = None
16
- item_based_knn_file_path: Optional[str] = None
17
- user_based_knn_file_path: Optional[str] = None
18
-
19
- @dataclass
20
- class ContentBasedModelArtifact:
21
  cosine_similarity_model_file_path:str
 
1
+ from dataclasses import dataclass
2
+ from typing import Optional
3
+
4
+ @dataclass
5
+ class DataIngestionArtifact:
6
+ feature_store_anime_file_path:str
7
+ feature_store_userrating_file_path:str
8
+
9
+ @dataclass
10
+ class DataTransformationArtifact:
11
+ merged_file_path:str
12
+
13
+ @dataclass
14
+ class CollaborativeModelArtifact:
15
+ svd_file_path:str
16
+ item_based_knn_file_path:str
17
+ user_based_knn_file_path:str
18
+
19
+ @dataclass
20
+ class ContentBasedModelArtifact:
21
  cosine_similarity_model_file_path:str