|
import sys |
|
from anime_recommender.loggers.logging import logging |
|
from anime_recommender.exception.exception import AnimeRecommendorException |
|
from anime_recommender.source.data_ingestion import DataIngestion |
|
from anime_recommender.entity.config_entity import TrainingPipelineConfig,DataIngestionConfig,DataTransformationConfig,CollaborativeModelConfig,ContentBasedModelConfig |
|
from anime_recommender.source.data_transformation import DataTransformation |
|
from anime_recommender.source.collaborative_recommender import CollaborativeModelTrainer |
|
from anime_recommender.source.content_based_recommender import ContentBasedModelTrainer |
|
from anime_recommender.source.top_anime_recommenders import PopularityBasedRecommendor |
|
|
|
|
|
if __name__ == "__main__": |
|
try: |
|
training_pipeline_config = TrainingPipelineConfig() |
|
data_ingestion_config = DataIngestionConfig(training_pipeline_config) |
|
data_ingestion = DataIngestion(data_ingestion_config) |
|
logging.info("Initiating Data Ingestion.") |
|
data_ingestion_artifact = data_ingestion.ingest_data() |
|
logging.info(f"Data ingestion completed.") |
|
print(data_ingestion_artifact) |
|
|
|
|
|
data_transformation_config = DataTransformationConfig(training_pipeline_config) |
|
data_transformation = DataTransformation(data_ingestion_artifact,data_transformation_config) |
|
logging.info("Initiating Data Transformation.") |
|
data_transformation_artifact = data_transformation.initiate_data_transformation() |
|
logging.info("Data Transformation Completed.") |
|
print(data_transformation_artifact) |
|
|
|
|
|
collaborative_model_trainer_config = CollaborativeModelConfig(training_pipeline_config) |
|
collaborative_model_trainer = CollaborativeModelTrainer(collaborative_model_trainer_config= collaborative_model_trainer_config,data_transformation_artifact=data_transformation_artifact) |
|
logging.info("Initiating Collaborative Model training.") |
|
collaborative_model_trainer_artifact = collaborative_model_trainer.initiate_model_trainer(model_type='user_knn') |
|
logging.info("Collaborative Model training completed.") |
|
print(collaborative_model_trainer_artifact) |
|
|
|
|
|
content_based_model_trainer_config = ContentBasedModelConfig(training_pipeline_config) |
|
content_based_model_trainer = ContentBasedModelTrainer(content_based_model_trainer_config=content_based_model_trainer_config,data_ingestion_artifact=data_ingestion_artifact) |
|
logging.info("Initiating Content Based Model training.") |
|
content_based_model_trainer_artifact = content_based_model_trainer.initiate_model_trainer() |
|
logging.info("Content Based Model training completed.") |
|
print(content_based_model_trainer_artifact) |
|
|
|
|
|
logging.info("Initiating Popularity based filtering.") |
|
filtering = PopularityBasedRecommendor(data_ingestion_artifact=data_ingestion_artifact) |
|
popularity_recommendations = filtering.initiate_model_trainer(filter_type='top_avg_rated') |
|
logging.info("Popularity based filtering completed.") |
|
|
|
except Exception as e: |
|
raise AnimeRecommendorException(e, sys) |
